Update app.py

app.py

@@ -9,8 +9,8 @@ model_id = "google/flan-t5-small"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Load tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("./hf-cache")
-model = AutoModelForSeq2SeqLM.from_pretrained("./hf-cache").to(device)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_id).to(device)
 
 # Define chatbot pipeline
 chatbot = pipeline("text2text-generation", model=model, tokenizer=tokenizer, device=0 if device == "cuda" else -1)
@@ -33,25 +33,28 @@ def get_nearby_businesses(user_location):
     try:
         businesses = find_businesses_by_location(user_location)
         if not businesses:
-            return "❌ No businesses found in your location."
+            return "<p>❌ No businesses found in your location.</p>"
         html = "<ul>"
         for biz in businesses:
-            html += f"<li><b>{biz['name']}</b><br>{biz['description']}<br>📍 {biz['location']} | 📞 {biz['contact']}</li>"
+            html += f"<li><b>{biz['name']}</b><br>{biz.get('description', '')}<br>📍 {biz['location']} | 📞 {biz['contact']}</li>"
         html += "</ul>"
         return html
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"<p>❌ Error: {str(e)}</p>"
 
 def submit_business(name, location, description, contact):
     if not (name and location and contact):
         return "❌ Please fill in all required fields."
-    insert_business({
-        "name": name,
-        "location": location,
-        "description": description,
-        "contact": contact
-    })
-    return "✅ Business submitted successfully!"
+    try:
+        insert_business({
+            "name": name,
+            "location": location,
+            "description": description,
+            "contact": contact
+        })
+        return "✅ Business submitted successfully!"
+    except Exception as e:
+        return f"❌ Error submitting business: {str(e)}"
 
 def get_location_name(coords):
     if not coords or len(coords) != 2:
@@ -59,12 +62,11 @@ def get_location_name(coords):
     lat, lon = coords
     return reverse_geocode(lat, lon)
 
-# Dummy map click function (since Gradio does not support map component directly)
 def update_location_from_coords(coord_str):
-    return coord_str  # Placeholder: In real app, use geolocation or map click
+    return coord_str  # Placeholder
 
 # --- Gradio Interface ---
-with gr.Blocks(title="Township Chatbot with GPS") as demo:
+with gr.Blocks() as demo:
     gr.Markdown("## 🛍 Township Business Chatbot Directory + GPS 🗺️")
 
     location_text = gr.Textbox(label="📍 Your Location", placeholder="e.g. Orange Farm, Gauteng")
@@ -89,4 +91,3 @@ with gr.Blocks(title="Township Chatbot with GPS") as demo:
         submit_btn.click(submit_business, [name, location_text, description, contact], submit_output)
 
 demo.launch()
-