Update app.py

app.py

@@ -9,63 +9,102 @@ import random
 GROQ_API_KEY = os.getenv("GROQ_API_KEY", "")
 client = Groq(api_key=GROQ_API_KEY)
 
-# Detect Language
+MODEL_ID = "llama-3.3-70b-versatile"
+
+# Multilingual UI setup
+translations = {
+    "en": {
+        "welcome": "Hey! I'm FlightAI. Tell me your travel plans or pick a button below!",
+        "questions": [
+            "Trip recommendations",
+            "Send me somewhere!",
+            "1-week planned vacations"
+        ],
+        "system_prompt": "You help users plan trips. Keep responses short and clear. If a user asks about booking a flight, hotel, or reservation, try to find a relevant link.",
+        "input_placeholder": "Type your travel question here..."
+    },
+    "fr": {
+        "welcome": "Salut ! Je suis FlightAI. Dis-moi ton projet de voyage ou choisis un bouton ci-dessous !",
+        "questions": [
+            "Recommandations de voyage",
+            "Envoyez-moi quelque part !",
+            "Voyage d'une semaine planifié"
+        ],
+        "system_prompt": "Vous aidez les utilisateurs à planifier des voyages. Réponses courtes et simples. Fournissez un lien si on vous demande une réservation.",
+        "input_placeholder": "Tapez votre question de voyage ici..."
+    },
+    "es": {
+        "welcome": "¡Hola! Soy FlightAI. Cuéntame tus planes de viaje o elige un botón a continuación.",
+        "questions": [
+            "Recomendaciones de viaje",
+            "¡Mándame a algún lugar!",
+            "Vacaciones planificadas de 1 semana"
+        ],
+        "system_prompt": "Ayudas a los usuarios a planificar viajes. Responde de manera clara y concisa. Si te preguntan sobre reservas, proporciona un enlace.",
+        "input_placeholder": "Escribe tu pregunta sobre viajes aquí..."
+    }
+}
+
 def detect_language(text):
+    """Detects whether the input text is in French, English, or Spanish."""
     try:
         lang = langdetect.detect(text)
-        return "fr" if lang == "fr" else "en"
+        return lang if lang in translations else "en"
     except:
         return "en"
 
-# Function to search for reservation links
+# Function to search for reservation websites
 def find_reservation_link(query):
+    """Finds a website link for reservations using a simple search API."""
     search_url = f"https://www.googleapis.com/customsearch/v1?q={query}+booking&key=your_google_api_key&cx=your_custom_search_engine_id"
     response = requests.get(search_url)
     data = response.json()
     
+    # Extract the first link
     if "items" in data and len(data["items"]) > 0:
         return data["items"][0]["link"]
     
     return "I couldn't find a link for that. Try searching manually."
 
-# Button Click Handler
-def handle_button_click(button_text):
+def handle_button_click(button_text, lang):
+    """Handles button clicks by sending predefined queries."""
     options = {
-        "Trip recommendations": "What are the top travel spots right now?",
-        "Send me somewhere!": f"Pick a place for me. Maybe {random.choice(['Paris', 'Tokyo', 'New York', 'Barcelona', 'Rome', 'Bali', 'Dubai', 'Sydney'])}!",
-        "1 week planned vacations": "Plan a 1-week trip for me."
+        translations[lang]["questions"][0]: "What are the top travel spots right now?",
+        translations[lang]["questions"][1]: f"Pick a place for me. Maybe {random.choice(['Paris', 'Tokyo', 'New York', 'Barcelona', 'Rome', 'Bali', 'Dubai', 'Sydney'])}!",
+        translations[lang]["questions"][2]: "Plan a 1-week trip for me."
     }
     return options.get(button_text, "")
 
-# Chat Function with Memory
-def chat_with_bot(message, history):
-    # Ensure history is correctly formatted
-    history = history or [{"role": "assistant", "content": "Hey! I'm FlightAI. How can I help with your travel plans?"}]
-
-    # If button text was used
-    if message in ["Trip recommendations", "Send me somewhere!", "1 week planned vacations"]:
-        message = handle_button_click(message)
+def chat_with_bot(message, history, lang="en"):
+    """Handles user queries and maintains chat memory."""
     
-    # Detect language
-    language = detect_language(message)
-
-    system_message = {
-        "en": "You help users plan trips. Keep responses short and clear. If a user asks about booking a flight, hotel, or reservation, try to find a link.",
-        "fr": "Vous aidez les utilisateurs à planifier des voyages. Réponses courtes et simples. Si un utilisateur demande une réservation, fournissez un lien."
-    }
+    # Detect language if not provided
+    if lang == "auto":
+        lang = detect_language(message)
+    
+    # If message is from a button click
+    if message in translations[lang]["questions"]:
+        message = handle_button_click(message, lang)
+    
+    # Convert history to AI-compatible format
+    messages = [{"role": "system", "content": translations[lang]["system_prompt"]}]
+    
+    for pair in history:
+        user_msg, bot_msg = pair
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
 
-    # Format history for chat completion
-    messages = [{"role": "system", "content": system_message[language]}] + history
+    # Add latest user message
     messages.append({"role": "user", "content": message})
 
-    # Check for reservation-related queries
+    # Check if the user is looking for reservations
     if any(keyword in message.lower() for keyword in ["book a flight", "hotel reservation", "car rental", "book a room"]):
         link = find_reservation_link(message)
-        return history + [{"role": "assistant", "content": f"Here's a link that might help: {link}"}]
+        return history + [(message, f"Here's a link that might help: {link}")]
 
     # Call AI model
     response = client.chat.completions.create(
-        model="llama-3.3-70b-versatile",
+        model=MODEL_ID,
         messages=messages,
         temperature=0.7,
         max_tokens=1024,
@@ -73,12 +112,10 @@ def chat_with_bot(message, history):
     )
 
     bot_reply = response.choices[0].message.content
-    history.append({"role": "user", "content": message})
-    history.append({"role": "assistant", "content": bot_reply})
-
+    history.append((message, bot_reply))
     return history
 
-# Gradio Interface
+# Gradio UI
 with gr.Blocks(css="""
     body { background-color: #A9B5DF; }
     .gradio-container { background-color: #A9B5DF; }
@@ -86,26 +123,45 @@ with gr.Blocks(css="""
     .gradio-button { background-color: #7886C7; color: white; border-radius: 10px; }
     .gradio-chatbot-message { background-color: #FFF2F2; color: #2D336B; padding: 10px; border-radius: 10px; }
 """) as demo:
-    gr.Markdown("# 🌍 ***FlightAI - Your Travel Assistant - Votre Assistant de Voyage*** ✈️\n")
+    gr.Markdown("# 🌍 ***FlightAI - Your Travel Assistant / Votre Assistant de Voyage / Tu Asistente de Viaje*** ✈️\n")
     
-    chatbot = gr.Chatbot(type="messages", value=[
-        {"role": "assistant", "content": "Hey! I'm FlightAI. Tell me your travel plans or pick a button below!"}
-    ])
+    # Language selector
+    language = gr.Radio(
+        choices=["English", "Français", "Español"], 
+        value="English", 
+        label="Language / Langue / Idioma"
+    )
+
+    chatbot = gr.Chatbot(type="messages", value=[{"role": "assistant", "content": translations["en"]["welcome"]}])
+    
+    state = gr.State([])  # Store conversation history
+    lang_code = gr.State("en")
 
     with gr.Row():
-        btn1 = gr.Button("Trip recommendations")
-        btn2 = gr.Button("Send me somewhere!")
-        btn3 = gr.Button("1 week planned vacations")
+        btn1 = gr.Button(translations["en"]["questions"][0])
+        btn2 = gr.Button(translations["en"]["questions"][1])
+        btn3 = gr.Button(translations["en"]["questions"][2])
 
-    user_input = gr.Textbox(placeholder="Type your travel question here...")
+    user_input = gr.Textbox(placeholder=translations["en"]["input_placeholder"])
+
+    # Function to update language settings
+    def update_ui_language(lang):
+        lang_code = {"English": "en", "Français": "fr", "Español": "es"}.get(lang, "en")
+        return lang_code, translations[lang_code]["questions"], translations[lang_code]["input_placeholder"], [{"role": "assistant", "content": translations[lang_code]["welcome"]}]
+
+    # Chat function
+    def submit_msg(message, history, lang):
+        return "", chat_with_bot(message, history, lang)
+
+    # Button click handlers
+    def button_click(button_text, history, lang):
+        return chat_with_bot(button_text, history, lang)
+
+    btn1.click(button_click, inputs=[btn1, state, lang_code], outputs=[chatbot])
+    btn2.click(button_click, inputs=[btn2, state, lang_code], outputs=[chatbot])
+    btn3.click(button_click, inputs=[btn3, state, lang_code], outputs=[chatbot])
+    user_input.submit(submit_msg, inputs=[user_input, state, lang_code], outputs=[chatbot])
+
+    language.change(update_ui_language, language, [lang_code, btn1, btn2, btn3, user_input, chatbot])
 
-    # Button Click Event Handling
-    def button_click(btn_text):
-        return chat_with_bot(btn_text, chatbot.value)
-    
-    btn1.click(button_click, inputs=[btn1], outputs=[chatbot])
-    btn2.click(button_click, inputs=[btn2], outputs=[chatbot])
-    btn3.click(button_click, inputs=[btn3], outputs=[chatbot])
-    user_input.submit(chat_with_bot, inputs=[user_input, chatbot], outputs=[chatbot])
-    
     demo.launch()