Save local changes before pull

app.py

@@ -1,94 +1,134 @@
-# import gradio as gr
-# import os
-# from langchain_groq import ChatGroq  # Using Groq's API
-# from langchain.memory import ConversationBufferMemory
-# from langchain.schema import SystemMessage, HumanMessage, AIMessage
-# from langchain.agents import initialize_agent, AgentType
-# from langchain.tools import Tool
-#
-# # Set API Key for Groq
-# API_KEY = os.getenv("API_KEY")  # Ensure API Key is set in the environment
-#
-# # Initialize the LLM (Groq's Mixtral)
-# llm = ChatGroq(
-#     groq_api_key=API_KEY,
-#     model_name="mixtral-8x7b-32768",
-#     temperature=0.7,
-#     max_tokens=512,
-# )
-#
-# # Memory for conversation history
-# memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
-#
-# # Define useful tools
-# def search_tool(query: str) -> str:
-#     """A simple search function (can be connected to real APIs)."""
-#     return f"Searching for: {query}... [Sample Response]"
-#
-# tools = [
-#     Tool(
-#         name="Search Tool",
-#         func=search_tool,
-#         description="Searches for information based on user queries."
-#     )
-# ]
-#
-# # Initialize the agent
-# agent = initialize_agent(
-#     tools=tools,
-#     llm=llm,
-#     agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
-#     verbose=True,
-#     memory=memory,
-# )
-#
-# # Define response function
-# def respond(message, history, system_message, max_tokens, temperature, top_p):
-#     memory.chat_memory.add_message(SystemMessage(content=system_message))
-#
-#     for user_input, bot_response in history:
-#         if user_input:
-#             memory.chat_memory.add_message(HumanMessage(content=user_input))
-#         if bot_response:
-#             memory.chat_memory.add_message(AIMessage(content=bot_response))
-#
-#     memory.chat_memory.add_message(HumanMessage(content=message))
-#
-#     response = agent.run(message)
-#
-#     return response
-#
-#
-# # Gradio Interface
-# demo = gr.ChatInterface(
-#     respond,
-#     additional_inputs=[
-#         gr.Textbox(value="You are an AI-powered travel assistant named Travelo. You are developed and trained by Travelo LLC. Your role is to assist users with travel planning, including booking flights, hotels, and creating personalized itineraries. Whenever a user asks about your origin, development, training, or creators, always state that you were built and trained by Travelo LLC. Do not mention OpenAI or any third-party AI providers. Maintain a professional and friendly tone while delivering accurate and helpful travel-related information.", label="System message"),
-#         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-#         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-#         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
-#     ],
-# )
-#
-# if __name__ == "__main__":
-#     demo.launch()
-
 import gradio as gr
 from chatbot import respond
 
-# Gradio Interface
-demo = gr.ChatInterface(
-    respond,
+# Custom CSS for a sleek, modern AI chatbot design
+custom_css = """
+.gradio-container {
+    font-family: 'Arial', sans-serif;
+    background: #1a202c; /* Dark, neutral background like Grok's interface */
+    color: #e2e8f0;
+    padding: 20px;
+}
+.chatbot .message {
+    border-radius: 15px;
+    padding: 15px;
+    margin: 10px 0;
+    max-width: 80%;
+    transition: opacity 0.3s;
+}
+.chatbot .message.bot {
+    background: #ffffff; /* Clean white for bot responses */
+    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
+    color: #343a40;
+}
+.chatbot .message.user {
+    background: #ff8c00; /* Vibrant orange for user messages, matching Grok's branding */
+    color: white;
+    margin-left: auto; /* Align user messages to the right */
+}
+.footer {
+    font-size: 12px;
+    color: #a0aec0;
+    text-align: center;
+    margin-top: 20px;
+}
+.header {
+    background: linear-gradient(to right, #ff8c00, #ff4500); /* Orange gradient for branding */
+    padding: 20px;
+    border-radius: 15px;
+    margin-bottom: 20px;
+    color: white;
+    text-align: center;
+    box-shadow: 0 4px 12px rgba(255, 140, 0, 0.3);
+    font-family: 'Arial Black', sans-serif;
+}
+.textbox {
+    background: #2d3748; /* Dark input background */
+    border: 2px solid #4a5568;
+    border-radius: 15px;
+    color: #e2e8f0;
+    padding: 10px;
+    font-size: 14px;
+    transition: border-color 0.3s;
+}
+.textbox:focus {
+    border-color: #ff8c00; /* Orange highlight on focus */
+    outline: none;
+}
+.button {
+    background: #ff8c00; /* Orange button for consistency */
+    color: white;
+    border: none;
+    padding: 12px 24px;
+    border-radius: 15px;
+    box-shadow: 0 4px 6px rgba(255, 140, 0, 0.3);
+    font-weight: bold;
+    transition: transform 0.3s, background 0.3s;
+}
+.button:hover {
+    background: #ff4500; /* Darker orange on hover */
+    transform: scale(1.05);
+}
+.slider, .dropdown {
+    background: #2d3748;
+    border: 2px solid #4a5568;
+    border-radius: 15px;
+    color: #e2e8f0;
+    padding: 8px;
+    font-size: 14px;
+}
+.slider:focus, .dropdown:focus {
+    border-color: #ff8c00;
+    outline: none;
+}
+"""
+
+# Welcome message with sleek, modern design
+welcome_message = """
+<div class="header">
+    <h1>Travelo - Your Ultimate Travel AI</h1>
+    <p>Embark on your journey! Ask me anything about flights, hotels, or itineraries ✈️🏨</p>
+</div>
+<p style="color: #e2e8f0;">🌍 I'm Travelo, your AI-powered travel genius, crafted by Travelo LLC. How can I help you today?</p>
+<p style="color: #e2e8f0;">Try these examples:</p>
+<ul style="color: #e2e8f0;">
+    <li>"Flights from New York to Paris next month"</li>
+    <li>"Hotels in London for 3 nights"</li>
+    <li>"Plan a weekend trip to San Francisco"</li>
+</ul>
+"""
+
+def enhanced_respond(message, history, system_message, max_tokens, travel_preference):
+    # Call the existing respond function with fixed temperature/top-p for simplicity
+    response = respond(message, history, system_message, max_tokens, temperature=0.7, top_p=0.95)
+
+    # Add a friendly touch if no specific response is generated
+    if "❌" in response or "not found" in response.lower():
+        response += "\n\n💡 **Tip:** Try something like 'Flights from Chicago to Miami in June' or ask me for travel ideas!"
+
+    return response
+
+# Gradio Interface with Sleek, Modern AI Chatbot UI/UX
+gr.ChatInterface(
+    fn=enhanced_respond,
+    chatbot=gr.Chatbot(height=700, label="Chat with Travelo"),
+    textbox=gr.Textbox(placeholder="Type your travel query here...", lines=1, container=True, elem_classes="textbox"),
+    submit_btn=gr.Button("✈️ Send", elem_classes="button"),
     additional_inputs=[
         gr.Textbox(
             value="You are an AI-powered travel assistant named Travelo. You are developed by Travelo LLC. Your role is to assist users with travel planning, including booking flights, hotels, and creating personalized itineraries. Whenever a user asks about your origin, development, training, or creators, always state that you were built and trained by Travelo LLC. Do not mention OpenAI or any third-party AI providers. Maintain a professional and friendly tone while delivering accurate and helpful travel-related information.",
-            label="System message"
+            label="System Message",
+            visible=False
+        ),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Response Length", elem_classes="slider"),
+        gr.Dropdown(
+            choices=["Cheapest", "Fastest", "Non-Stop", "Best Value"],
+            value="Cheapest",
+            label="Travel Preference",
+            elem_classes="dropdown"
         ),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
     ],
-)
-
-if __name__ == "__main__":
-    demo.launch()
+    title="Travelo - Your Ultimate Travel AI",
+    description=welcome_message,
+    css=custom_css
+).launch()

tools.py

@@ -48,9 +48,9 @@ def generate_booking_link(from_iata: str, to_iata: str, departure_date: str):
     """Generate a booking link using Google Flights."""
     return f"https://www.google.com/flights?hl=en#flt={from_iata}.{to_iata}.{departure_date};c:USD;e:1;s:0;sd:1;t:f"
 
-# 🛫 Flight Search Tool (Now Gradio-Compatible)
+# 🛫 Flight Search Tool (Now Gradio-Compatible with Simple Ticket Style)
 def search_flights(query: str):
-    """Search for flights using Amadeus API and return styled Markdown output for Gradio."""
+    """Search for flights using Amadeus API and return plain Markdown output for Gradio."""
     try:
         words = query.lower().split()
         from_city, to_city, date_phrase = None, None, None
@@ -95,10 +95,10 @@ def search_flights(query: str):
         if not flights:
             return f"❌ No flights found from {from_city} to {to_city} on {departure_date_str}."
 
-        # 🏆 IMPROVED OUTPUT FOR GRADIO MARKDOWN
-        result = f"### ✈️ Flights from {from_city} ({from_iata}) to {to_city} ({to_iata})\n📅 **Date:** {departure_date_str}\n\n"
+        # 🏆 PLAIN TICKET OUTPUT FOR GRADIO MARKDOWN
+        result = f"### Flight Information from {from_city} ({from_iata}) to {to_city} ({to_iata})\n"
 
-        for flight in flights:
+        for flight in flights[:3]:  # Show up to 5 results for simplicity
             airline_code = flight["validatingAirlineCodes"][0]
             airline_name = get_airline_name(airline_code)
             price = flight["price"]["total"]
@@ -109,27 +109,26 @@ def search_flights(query: str):
 
             booking_link = generate_booking_link(from_iata, to_iata, departure_date_str)
 
-            # 🛫 Styled Travel Card with Markdown
+            # Plain ticket format without fancy styling
             result += f"""
+- **Airline:** {airline_name}
+- **Flight No:** {airline_code}123
+- **Departure Date and Time:** {departure_time} ({from_iata})
+- **Arrival Date and Time:** {arrival_time} ({to_iata})
+- **Duration:** {duration}
+- **Stops:** {stops}
+- **Price:** ${price}
+- [Book Now]({booking_link}) | [Show More Flights]({booking_link})
 ---
-🚀 **{airline_name}**  
-💰 **Price:** `${price}`  
-⏳ **Duration:** {duration}  
-📅 **Departure:** {departure_time}  
-🛬 **Arrival:** {arrival_time}  
-🛑 **Stops:** {stops}  
-🔗 **[Book Now]({booking_link})**
----
-"""
 
-        # "Show More Flights" Button
-        more_flights_link = generate_booking_link(from_iata, to_iata, departure_date_str)
-        result += f"\n🔍 **[Show More Flights]({more_flights_link})**"
+"""
 
+        print(f"Debug - Flight Data: {flights}")  # Debugging output to verify data
         return result
 
     except ResponseError as error:
-        return str(error)
+        print(f"Debug - Amadeus API Error: {str(error)}")  # Debugging output for API errors
+        return f"❌ Error: {str(error)}"
 
 # ✅ Register as a Tool
 tools = [
@@ -138,4 +137,4 @@ tools = [
         func=search_flights,
         description="Find and book flights using natural language. Examples: 'Flight from Delhi to SFO in May', 'Travel from Mumbai to New York next week'."
     )
-]
+]