Update app.py

app.py

@@ -8,106 +8,101 @@ genai.configure(api_key=GEMINI_KEY)
 model = genai.GenerativeModel('gemini-pro')
 
 CSS = """
-body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
-.header { text-align: center; margin-bottom: 30px; }
-.input-section { display: flex; gap: 10px; margin-bottom: 20px; }
-#question { flex-grow: 1; padding: 10px; }
-.category { margin: 25px 0; border-bottom: 2px solid #eee; padding-bottom: 20px; }
-.category-title { color: #2c3e50; font-size: 1.4em; margin-bottom: 15px; }
-.item-card { background: white; padding: 15px; border-radius: 8px; margin: 10px 0; box-shadow: 0 2px 4px rgba(0,0,0,0.1); }
-.item-card h3 { margin: 0 0 8px 0; color: #3498db; }
-.link { color: #2980b9; text-decoration: underline; }
+body { background: #f5f7fa; font-family: 'Arial', sans-serif; }
+.chat-container { max-width: 800px; margin: 0 auto; }
+.msg-user { background: #e3f2fd; padding: 15px; border-radius: 15px; margin: 10px; max-width: 70%; float: right; }
+.msg-bot { background: white; padding: 15px; border-radius: 15px; margin: 10px; max-width: 70%; box-shadow: 0 2px 4px rgba(0,0,0,0.1); }
+.card { border-left: 4px solid #2196F3; padding: 10px; margin: 10px 0; }
+.price { color: #2e7d32; font-weight: bold; }
+.link { color: #1565C0; text-decoration: underline; }
+.reset-btn { margin-top: 20px; }
 """
 
-def parse_response(response):
-    categories = {}
-    current_category = None
+class ChatManager:
+    def __init__(self):
+        self.context = {"city": None}
     
-    # Dividi in categorie
-    lines = response.split('\n')
-    for line in lines:
-        if line.startswith('## '):
-            current_category = line.replace('## ', '').strip()
-            categories[current_category] = []
-        elif line.startswith('- **') or line.startswith('1. **'):
-            parts = re.split(r'\*\*|\*', line)
-            name = parts[1].strip()
-            desc = parts[-1].strip()
-            
-            # Estrai link
-            links = re.findall(r'(https?://\S+)', desc)
-            desc = re.sub(r'(https?://\S+)', '', desc).strip()
-            
-            categories[current_category].append({
-                "name": name,
-                "description": desc,
-                "links": links
-            })
+    def extract_city(self, text):
+        cities = re.findall(r'\b(?:a|in|su|per)\s+([A-Z][a-z]+(?:\s+[A-Z][a-z]+)*)\b', text, re.IGNORECASE)
+        return cities[0] if cities else self.context["city"]
     
-    return categories
-
-def generate_response(question):
-    # Controlla se è una richiesta specifica
-    specific_query = re.search(r'\b(dove|cosa|come|quale|quando)\b', question, re.IGNORECASE)
-    
-    if specific_query:
-        # Modalità risposta libera
-        response = model.generate_content(f"Rispondi in italiano in formato testo semplice: {question}").text
-        return f"<div class='free-response'>{response}</div>"
-    else:
-        # Modalità strutturata
-        city = re.search(r'\b(a|in|per|su)\s+([A-Za-z\s]+)', question, re.IGNORECASE)
-        city = city.group(2).strip() if city else question
+    def format_response(self, response):
+        structured = []
+        current_item = {}
         
-        prompt = f"""
-        Genera una guida turistica per {city} in italiano con:
-        - 3-5 luoghi da vedere con breve descrizione e sito ufficiale se disponibile
-        - 3 ristoranti tipici con specialità e eventuale sito web
-        - 3 opzioni alloggio con range prezzi
-        - 2-3 attività caratteristiche
-        - Formatta in markdown con categorie ##
-        - Includi link dove appropriato
-        """
+        lines = response.split('\n')
+        for line in lines:
+            if line.startswith('- **'):
+                parts = re.split(r'\*\*:?|\*', line)
+                if len(parts) > 2:
+                    key = parts[1].strip().lower()
+                    value = parts[2].strip()
+                    current_item[key] = value
+            elif line.strip() == '' and current_item:
+                structured.append(current_item)
+                current_item = {}
         
-        response = model.generate_content(prompt).text
-        categories = parse_response(response)
+        if current_item:
+            structured.append(current_item)
         
-        html_output = ""
-        for category, items in categories.items():
-            section = f"""
-            <div class='category'>
-                <div class='category-title'>{category}</div>
+        html = ""
+        for item in structured:
+            html += f"""
+            <div class='card'>
+                <h3>{item.get('nome', '')}</h3>
+                <p>{item.get('descrizione', '')}</p>
+                <p class='price'>{item.get('prezzo', '')}</p>
+                {f"<a href='{item['link']}' class='link' target='_blank'>Maggiori informazioni</a>" if 'link' in item else ''}
+            </div>
             """
-            
-            for item in items:
-                links = "".join([f"<a href='{link}' class='link' target='_blank'>Sito ufficiale</a>" for link in item['links']])
-                
-                section += f"""
-                <div class='item-card'>
-                    <h3>{item['name']}</h3>
-                    <p>{item['description']}</p>
-                    {links}
-                </div>
-                """
-            
-            section += "</div>"
-            html_output += section
-        
-        return html_output
+        return html if html else response
+
+chat_manager = ChatManager()
+
+def respond(message, history):
+    # Estrai e mantieni contesto città
+    city = chat_manager.extract_city(message)
+    if city:
+        chat_manager.context["city"] = city
+    
+    if not chat_manager.context["city"]:
+        return "Per favore dimmi prima di quale città vuoi informazioni! 🌍"
+    
+    # Genera risposta
+    prompt = f"""
+    Sei un esperto guida turistica per {chat_manager.context["city"]}. 
+    Rispondi in italiano strutturando ogni suggerimento con:
+    - **Nome**: [nome luogo]
+    - **Descrizione**: [breve descrizione 20-30 parole]
+    - **Prezzo**: [se applicabile]
+    - **Link**: [sito ufficiale se disponibile]
+    
+    Richiesta utente: {message}
+    """
+    
+    try:
+        response = model.generate_content(prompt).text
+        return chat_manager.format_response(response)
+    except Exception as e:
+        return f"⚠️ Errore: {str(e)}"
+
+def reset_context():
+    chat_manager.context = {"city": None}
+    return "Contesto resettato. Di quale città vuoi parlare ora? 🌆"
 
 with gr.Blocks(css=CSS) as app:
-    gr.HTML("<div class='header'><h1>🌍 Guida Turistica Digitale</h1><p>Chiedimi informazioni su qualsiasi destinazione!</p></div>")
+    gr.Markdown("# 🗺️ Assistente Turistico Chat")
+    
+    chatbot = gr.Chatbot(label="Chat", elem_classes="chat-container")
+    msg = gr.Textbox(label="Scrivi il tuo messaggio...")
     
-    with gr.Row(equal_height=True, variant="panel"):
-        question = gr.Textbox(placeholder="Es: Cosa visitare a Roma?", elem_id="question")
-        submit_btn = gr.Button("Cerca", variant="primary")
+    with gr.Row():
+        send_btn = gr.Button("Invia")
+        reset_btn = gr.Button("Cambia Città", variant="secondary")
     
-    output = gr.HTML()
+    send_btn.click(respond, [msg, chatbot], [msg, chatbot])
+    reset_btn.click(reset_context, None, [chatbot])
     
-    submit_btn.click(
-        fn=generate_response,
-        inputs=question,
-        outputs=output
-    )
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
 app.launch()