Create app.py

app.py

@@ -0,0 +1,105 @@
+import os
+import gradio as gr
+from langchain.agents import AgentExecutor, Tool
+from langchain.prompts import PromptTemplate
+from langchain_huggingface import HuggingFaceEndpoint
+from dotenv import load_dotenv  # Optional for .env files
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Get Hugging Face token from environment variable
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("Hugging Face API token not found. Set HF_TOKEN in your environment variables.")
+
+# Define the Hugging Face model endpoint
+llm = HuggingFaceEndpoint(
+    repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+    huggingfacehub_api_token=HF_TOKEN.strip(),
+    temperature=0.7,
+    max_new_tokens=150,
+)
+
+# Define specialized agents
+
+def flight_booking(input):
+    return f"Searching flights for: {input}"
+
+flight_tool = Tool(
+    name="Flight Booking Tool",
+    func=flight_booking,
+    description="Finds and books airline flights"
+)
+
+def hotel_booking(input):
+    return f"Searching hotels for: {input}"
+
+hotel_tool = Tool(
+    name="Hotel Booking Tool",
+    func=hotel_booking,
+    description="Searches and books accommodations"
+)
+
+def transportation_booking(input):
+    return f"Arranging transportation for: {input}"
+
+transport_tool = Tool(
+    name="Transportation Booking Tool",
+    func=transportation_booking,
+    description="Handles rental cars, shuttles, or trains"
+)
+
+def activity_booking(input):
+    return f"Finding activities for: {input}"
+
+activity_tool = Tool(
+    name="Activity Booking Tool",
+    func=activity_booking,
+    description="Books activities, tours, and events"
+)
+
+tools = [flight_tool, hotel_tool, transport_tool, activity_tool]
+
+# Define a prompt for the planner agent
+planner_prompt = PromptTemplate(
+    template="You are a travel planner. Your task is to coordinate tools to plan a trip based on user input: {input}",
+    input_variables=["input"]
+)
+
+# Create the agent
+planner_agent = AgentExecutor.from_agent_and_tools(
+    llm=llm,
+    tools=tools,
+    verbose=True
+)
+
+# Define the Gradio function
+def plan_trip(user_input):
+    """
+    Processes user input through the planner agent and returns the output.
+    """
+    try:
+        output = planner_agent.run(user_input)
+        return output
+    except Exception as e:
+        return f"Error: {e}"
+
+# Create the Gradio interface
+with gr.Blocks() as travel_planner_app:
+    gr.Markdown("## AI Travel Planner")
+    gr.Markdown("Enter your trip details below, and the AI will help you plan your travel itinerary!")
+    
+    with gr.Row():
+        user_input = gr.Textbox(
+            label="Trip Details",
+            placeholder="E.g., Plan a trip to Paris from May 10 to May 15 for a family of 4."
+        )
+    submit_button = gr.Button("Plan Trip")
+    output = gr.Textbox(label="Travel Plan")
+    
+    submit_button.click(plan_trip, inputs=user_input, outputs=output)
+
+# Launch the Gradio app
+if __name__ == "__main__":
+    travel_planner_app.launch()