Update app.py

app.py

@@ -1,70 +1,83 @@
 import gradio as gr
 
-# Ububiko bw'intangiriro
+# Stock n'ibiciro (price ku kintu kimwe)
 stock = {
-    "Potatoes": 100,
-    "Tomatoes": 50,
-    "Carrots": 30,
+    "Ibiryayi": 100,
+    "Ibitunguru": 50,
+    "Karoti": 30,
     "Ibicuruzwa_A": 50,
     "Ibicuruzwa_B": 8,
     "Ibicuruzwa_C": 0,
     "Ibicuruzwa_D": 15,
+    "Amashu": 40,
+    "Imyembe": 25,
+    "Inanasi": 60,
+    "Amashaza": 70,
+    "Ibinyomoro": 10,
+    # ... wakomeza ugashyiramo ibindi ukoresha stock yawe ...
 }
 
-def add_products_batch(batch_input):
-    """
-    Buri murongo wanditse muri format: izina, quantity
-    Urugero:
-    Potatoes, 100
-    Tomatoes, 50
-    """
-    if not batch_input.strip():
-        return "Andika ibicuruzwa n'ububiko bya byo, buri kimwe ku murongo, mu format: izina, quantity.", gr.Dropdown.update(choices=list(stock.keys()))
-    
-    lines = batch_input.strip().split("\n")
-    messages = []
-    for line in lines:
-        parts = line.split(",")
-        if len(parts) != 2:
-            messages.append(f"Umurongo '{line}' ntusobanutse neza. Format: izina, quantity")
-            continue
-        product_name = parts[0].strip()
-        try:
-            quantity = int(parts[1].strip())
-            if quantity < 0:
-                messages.append(f"Ingano y'ububiko igomba kuba itari hasi ya zeru: '{line}'")
-                continue
-        except ValueError:
-            messages.append(f"Ingano y'ububiko si integer: '{line}'")
-            continue
-        
-        stock[product_name] = stock.get(product_name, 0) + quantity
-        messages.append(f"'{product_name}' yongewemo/ yavuguruwe: {stock[product_name]}")
-    
-    return "\n".join(messages), gr.Dropdown.update(choices=list(stock.keys()))
+prices = {
+    "Ibiryayi": 200,         # price mu mafaranga y'u Rwanda (RWF)
+    "Ibitunguru": 150,
+    "Karoti": 120,
+    "Ibicuruzwa_A": 500,
+    "Ibicuruzwa_B": 800,
+    "Ibicuruzwa_C": 1000,
+    "Ibicuruzwa_D": 400,
+    "Amashu": 250,
+    "Imyembe": 300,
+    "Inanasi": 350,
+    "Amashaza": 180,
+    "Ibinyomoro": 600,
+    # ... na hano ukoresha prices ku bicuruzwa byose ...
+}
 
-def get_stock_value(selected_product):
-    if selected_product not in stock:
-        return "Icyo gicuruzwa ntikirimo mu bubiko."
-    return f"Ububiko bwa '{selected_product}' ni: {stock[selected_product]}"
+def show_price(product_name):
+    product_name = product_name.strip()
+    if product_name == "":
+        return "Hitamo igicuruzwa."
+    if product_name not in stock or product_name not in prices:
+        return f"Icyo gicuruzwa '{product_name}' ntikirimo mu bubiko cyangwa nta giciro gifite."
+    price = prices[product_name]
+    quantity = stock[product_name]
+    return (f"Igicuruzwa '{product_name}' gifite ububiko bungana na {quantity} kandi igiciro cya kimwe ni {price} RWF.")
+
+def calc_total(product_name, quantity):
+    product_name = product_name.strip()
+    if product_name == "":
+        return "Andika cyangwa hitamo igicuruzwa."
+    if product_name not in stock or product_name not in prices:
+        return f"Icyo gicuruzwa '{product_name}' ntikirimo mu bubiko cyangwa nta giciro gifite."
+    if quantity < 0:
+        return "Ingano y'ububiko igomba kuba itari hasi ya zeru."
+    
+    available = stock[product_name]
+    price = prices[product_name]
+    total_price = price * quantity
+    return (f"Ububiko bwa '{product_name}' ni {available}. "
+            f"Uhisemo kugura/ kongera quantity {quantity}. "
+            f"Igiciro cyose: {total_price} RWF.")
 
 with gr.Blocks() as demo:
-    gr.Markdown("# AI - Gucunga Ububiko ku Isoko")
+    gr.Markdown("# Gucunga Ububiko & Kwerekana Ibiciro")
     
-    with gr.Column():
-        gr.Markdown("### Ongera ibicuruzwa byinshi icyarimwe (buri kimwe ku murongo: izina, quantity)")
-        batch_input = gr.Textarea(label="Andika ibicuruzwa na quantity (buri murongo: izina, quantity)", lines=5)
-        add_btn = gr.Button("Ongeramo ibicuruzwa byinshi")
-        output_add = gr.Textbox(label="Ibisubizo byo kongera ububiko")
+    with gr.Row():
+        product_dropdown = gr.Dropdown(choices=list(stock.keys()), label="Hitamo igicuruzwa")
+        show_price_btn = gr.Button("Erekana igiciro n'ububiko")
+        price_output = gr.Textbox(label="Ibisubizo")
 
-        add_btn.click(add_products_batch, inputs=batch_input, outputs=[output_add, batch_input])
+    show_price_btn.click(show_price, inputs=product_dropdown, outputs=price_output)
 
-        gr.Markdown("### Reba ububiko bw'igicuruzwa runaka")
-        product_dropdown = gr.Dropdown(choices=list(stock.keys()), label="Hitamo igicuruzwa")
-        check_btn = gr.Button("Reba ububiko")
-        output_check = gr.Textbox(label="Agaciro k'ububiko")
+    with gr.Row():
+        product_input = gr.Textbox(label="Andika cyangwa hitamo igicuruzwa", value="")
+        quantity_input = gr.Number(label="Andika ingano yifuza", value=0, precision=0)
+        calc_btn = gr.Button("Bara igiciro rusange")
+        total_output = gr.Textbox(label="Ibisubizo")
 
-        check_btn.click(get_stock_value, inputs=product_dropdown, outputs=output_check)
+    calc_btn.click(calc_total, inputs=[product_input, quantity_input], outputs=total_output)
 
 if __name__ == "__main__":
     demo.launch()
+
+