Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 import datetime
-from typing import Dict, List, Tuple
+from typing import Dict, List
 import random
 from huggingface_hub import InferenceClient
 
+# Initialize the Hugging Face client
+client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+
 class TravelPlanner:
     def __init__(self):
         self.accommodation_types = {
@@ -34,14 +37,6 @@ class TravelPlanner:
                           budget_level: str) -> List[Dict]:
         """
         Generate a daily itinerary based on destination and budget level
-        
-        Args:
-            destination: Name of the destination
-            num_days: Number of days for the trip
-            budget_level: luxury/mid_range/budget/hostel
-            
-        Returns:
-            List of daily activities and estimated times
         """
         itinerary = []
         
@@ -74,37 +69,24 @@ class TravelPlanner:
                         num_people: int) -> Dict:
         """
         Calculate estimated budget based on duration and comfort level
-        
-        Args:
-            num_days: Number of days for the trip
-            budget_level: luxury/mid_range/budget/hostel
-            num_people: Number of travelers
-            
-        Returns:
-            Dictionary with budget breakdown
         """
-        # Get cost ranges based on budget level
         accommodation_range = self.accommodation_types[budget_level]
         meal_range = self.meal_costs[budget_level]
         
-        # Calculate daily costs
         daily_accommodation = random.uniform(*accommodation_range)
         daily_meals = random.uniform(*meal_range) * 3  # 3 meals per day
         daily_activities = random.uniform(30, 100)  # Average activity cost
         daily_transport = random.uniform(10, 50)    # Local transport
         
-        # Calculate total costs
         total_accommodation = daily_accommodation * num_days
         total_meals = daily_meals * num_days
         total_activities = daily_activities * num_days
         total_transport = daily_transport * num_days
         
-        # Multiply by number of people (except accommodation which is per room)
         total_meals *= num_people
         total_activities *= num_people
         total_transport *= num_people
         
-        # Add 10% buffer for miscellaneous expenses
         buffer = (total_accommodation + total_meals + total_activities + total_transport) * 0.1
         
         total_cost = total_accommodation + total_meals + total_activities + total_transport + buffer
@@ -124,14 +106,6 @@ class TravelPlanner:
                      budget: Dict) -> str:
         """
         Format the itinerary and budget into a readable string
-        
-        Args:
-            destination: Name of the destination
-            itinerary: List of daily activities
-            budget: Dictionary of budget breakdown
-            
-        Returns:
-            Formatted string with itinerary and budget
         """
         output = f"Travel Plan for {destination}\n\n"
         output += "=== ITINERARY ===\n\n"
@@ -153,66 +127,75 @@ class TravelPlanner:
         
         return output
 
-def generate_travel_plan(destination: str,
-                        num_days: int,
-                        budget_level: str,
-                        num_people: int) -> str:
-    """
-    Main function to generate complete travel plan
-    
-    Args:
-        destination: Name of the destination
-        num_days: Number of days for the trip
-        budget_level: luxury/mid_range/budget/hostel
-        num_people: Number of travelers
-        
-    Returns:
-        Formatted string with complete travel plan
+def respond(
+    message: str,
+    history: List[Dict[str, str]],  # Updated to use dict format
+    system_message: str,
+    max_tokens: int,
+    temperature: float,
+    top_p: float
+) -> str:
     """
-    planner = TravelPlanner()
-    itinerary = planner.generate_itinerary(destination, num_days, budget_level)
-    budget = planner.calculate_budget(num_days, budget_level, num_people)
-    return planner.format_output(destination, itinerary, budget)
-
-# Integration with your Gradio interface
-def respond_with_travel_plan(message: str,
-                           history: List[Tuple[str, str]],
-                           system_message: str,
-                           max_tokens: int,
-                           temperature: float,
-                           top_p: float) -> str:
+    Process chat message and generate travel plan if requested
     """
-    Modified respond function to handle travel planning requests
-    """
-    # Parse the input message for travel parameters
-    # This is a simple example - you might want to add more sophisticated parsing
-    try:
-        # Example message format: "Plan a trip to Paris for 5 days, mid_range budget, 2 people"
-        parts = message.lower().split()
-        destination_idx = parts.index("to") + 1
-        days_idx = parts.index("days") - 1
-        budget_idx = parts.index("budget") - 1
-        people_idx = parts.index("people") - 1
-        
-        destination = parts[destination_idx].capitalize()
-        num_days = int(parts[days_idx])
-        budget_level = parts[budget_idx]
-        num_people = int(parts[people_idx])
-        
-        # Generate travel plan
-        response = generate_travel_plan(destination, num_days, budget_level, num_people)
-        return response
-        
-    except (ValueError, IndexError):
-        return "I couldn't understand your travel request. Please use the format: 'Plan a trip to [destination] for [X] days, [budget_level] budget, [X] people'"
+    # Check if this is a travel planning request
+    if "plan a trip" in message.lower():
+        try:
+            # Parse the message
+            parts = message.lower().split()
+            destination_idx = parts.index("to") + 1
+            days_idx = parts.index("days") - 1
+            budget_idx = parts.index("budget") - 1
+            people_idx = parts.index("people") - 1
+            
+            destination = parts[destination_idx].capitalize()
+            num_days = int(parts[days_idx])
+            budget_level = parts[budget_idx]
+            num_people = int(parts[people_idx])
+            
+            # Generate travel plan
+            planner = TravelPlanner()
+            itinerary = planner.generate_itinerary(destination, num_days, budget_level)
+            budget = planner.calculate_budget(num_days, budget_level, num_people)
+            return planner.format_output(destination, itinerary, budget)
+            
+        except (ValueError, IndexError):
+            return "I couldn't understand your travel request. Please use the format: 'Plan a trip to [destination] for [X] days, [budget_level] budget, [X] people'"
+    
+    # If not a travel request, use the Hugging Face model
+    messages = [{"role": "system", "content": system_message}]
+    
+    # Convert history to the correct format
+    for msg in history:
+        messages.append({"role": msg["role"], "content": msg["content"]})
+    
+    messages.append({"role": "user", "content": message})
+    
+    response = ""
+    for token in client.chat_completion(
+        messages,
+        max_tokens=max_tokens,
+        stream=True,
+        temperature=temperature,
+        top_p=top_p,
+    ):
+        response += token.choices[0].delta.content
+    return response
 
-# Update your main Gradio interface
+# Create the Gradio interface with updated message format
 demo = gr.ChatInterface(
-    respond_with_travel_plan,
+    respond,
     additional_inputs=[
-        gr.Textbox(value="You are a travel planning assistant.", label="System message"),
+        gr.Textbox(value="You are a travel planning assistant who can also chat about other topics.", 
+                  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)"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, 
+                 label="Top-p (nucleus sampling)"),
     ],
-)
+    message_history=True,  # Enable message history
+)
+
+# Launch the application
+if __name__ == "__main__":
+    demo.launch()