Update app.py

app.py

@@ -14,11 +14,10 @@ stock = {
     "Inanasi": 60,
     "Amashaza": 70,
     "Ibinyomoro": 10,
-    # ... wakomeza ugashyiramo ibindi ukoresha stock yawe ...
 }
 
 prices = {
-    "Ibiryayi": 200,         # price mu mafaranga y'u Rwanda (RWF)
+    "Ibiryayi": 200,
     "Ibitunguru": 150,
     "Karoti": 120,
     "Ibicuruzwa_A": 500,
@@ -30,33 +29,31 @@ prices = {
     "Inanasi": 350,
     "Amashaza": 180,
     "Ibinyomoro": 600,
-    # ... na hano ukoresha prices ku bicuruzwa byose ...
 }
 
 def show_price(product_name):
-    product_name = product_name.strip()
-    if product_name == "":
+    if not product_name:
         return "Hitamo igicuruzwa."
-    if product_name not in stock or product_name not in prices:
+    price = prices.get(product_name)
+    quantity = stock.get(product_name)
+    if price is None or quantity is None:
         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.")
+    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 not 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."
-    if quantity < 0:
-        return "Ingano y'ububiko igomba kuba itari hasi ya zeru."
-    
+    if quantity is None or quantity <= 0:
+        return "Andika ingano yifuza kugura (ikwiye kuba irenze 0)."
     available = stock[product_name]
     price = prices[product_name]
+    if quantity > available:
+        return f"Ububiko bwa '{product_name}' buboneka ni {available}, ntabwo ushobora kugura {quantity}."
     total_price = price * quantity
     return (f"Ububiko bwa '{product_name}' ni {available}. "
-            f"Uhisemo kugura/ kongera quantity {quantity}. "
+            f"Uhisemo kugura quantity {quantity}. "
             f"Igiciro cyose: {total_price} RWF.")
 
 with gr.Blocks() as demo:
@@ -65,17 +62,17 @@ with gr.Blocks() as demo:
     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")
+        price_output = gr.Textbox(label="Ibisubizo", interactive=False)
 
     show_price_btn.click(show_price, inputs=product_dropdown, outputs=price_output)
 
     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)
+        product_dropdown2 = gr.Dropdown(choices=list(stock.keys()), label="Hitamo igicuruzwa (kubara igiciro rusange)")
+        quantity_input = gr.Number(label="Andika ingano yifuza kugura", value=1, precision=0)
         calc_btn = gr.Button("Bara igiciro rusange")
-        total_output = gr.Textbox(label="Ibisubizo")
+        total_output = gr.Textbox(label="Ibisubizo", interactive=False)
 
-    calc_btn.click(calc_total, inputs=[product_input, quantity_input], outputs=total_output)
+    calc_btn.click(calc_total, inputs=[product_dropdown2, quantity_input], outputs=total_output)
 
 if __name__ == "__main__":
     demo.launch()