Update app.py

app.py

@@ -1,28 +1,44 @@
 import gradio as gr
 
-# Sample stock data
-stock = {
-    "Ibicuruzwa_A": 50,
-    "Ibicuruzwa_B": 8,
-    "Ibicuruzwa_C": 0,
-    "Ibicuruzwa_D": 15,
-}
+# 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."
+    
+    # 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."
 
-# Interface ya Gradio
-iface = gr.Interface(
-    fn=check_stock,
-    inputs=gr.Slider(minimum=0, maximum=50, step=1, label="Hitamo umupaka w'ububiko uri hasi cyane"),
-    outputs="text",
-    title="AI - Gucunga Ububiko ku Isoko",
-    description="Hitamo umupaka w'ububiko kugirango urebe ibicuruzwa bigomba kongerwa."
-)
+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)
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()