Update app.py

app.py

@@ -1,44 +1,70 @@
 import gradio as gr
 
-# Fungura stock nshya (nta na kimwe gihari mu ntangiriro)
-stock = {}
-
-def add_product(product_name, quantity):
-    if product_name.strip() == "":
-        return "Andika izina ry'igicuruzwa."
-    if quantity < 0:
-        return "Ingano y'ububiko igomba kuba itari hasi ya zeru."
+# Ububiko bw'intangiriro
+stock = {
+    "Potatoes": 100,
+    "Tomatoes": 50,
+    "Carrots": 30,
+    "Ibicuruzwa_A": 50,
+    "Ibicuruzwa_B": 8,
+    "Ibicuruzwa_C": 0,
+    "Ibicuruzwa_D": 15,
+}
+
+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]}")
     
-    # Ongeraho cyangwa uvugurure ububiko
-    stock[product_name] = stock.get(product_name, 0) + quantity
-    return f"Ibicuruzwa '{product_name}' byongeweho/ byavuguruwe. Ububiko ubu: {stock[product_name]}"
-
-def check_stock(threshold):
-    if not stock:
-        return "Nta bicuruzwa biri mu bubiko."
-    low_stock_items = [f"{item} (Ububiko: {qty})" for item, qty in stock.items() if qty < threshold]
-    if low_stock_items:
-        return "Ibi bicuruzwa biri mu bubiko buke, hasabwa kongera:\n" + "\n".join(low_stock_items)
-    else:
-        return "Ububiko bwose burasa neza! Nta kibazo cyo kongera ububiko."
+    return "\n".join(messages), gr.Dropdown.update(choices=list(stock.keys()))
+
+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]}"
 
 with gr.Blocks() as demo:
     gr.Markdown("# AI - Gucunga Ububiko ku Isoko")
     
-    with gr.Row():
-        product_name = gr.Textbox(label="Izina ry'igicuruzwa")
-        quantity = gr.Number(label="Ingano y'ububiko (positive integer)", value=0, precision=0)
-        add_btn = gr.Button("Ongeraho / Vugurura ububiko")
-    
-    output_add = gr.Textbox(label="Ibisubizo byo kongera ububiko")
-    
-    add_btn.click(add_product, inputs=[product_name, quantity], outputs=output_add)
-    
-    threshold = gr.Slider(0, 50, value=10, step=1, label="Hitamo umupaka w'ububiko uri hasi cyane")
-    check_btn = gr.Button("Genura ububiko buke")
-    output_check = gr.Textbox(label="Ibisubizo byo kugenzura ububiko")
-    
-    check_btn.click(check_stock, inputs=threshold, outputs=output_check)
+    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")
+
+        add_btn.click(add_products_batch, inputs=batch_input, outputs=[output_add, batch_input])
+
+        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")
+
+        check_btn.click(get_stock_value, inputs=product_dropdown, outputs=output_check)
 
 if __name__ == "__main__":
     demo.launch()