|
|
import gradio as gr |
|
|
from transformers import pipeline |
|
|
|
|
|
|
|
|
chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1") |
|
|
|
|
|
def generate_itinerary(destination, start_date, end_date, traveler_type, companion): |
|
|
"""Generates an AI-powered travel itinerary based on user preferences""" |
|
|
|
|
|
prompt = f""" |
|
|
You are a travel assistant. Generate a detailed day-by-day itinerary for a trip to {destination}. |
|
|
- Trip Duration: {start_date} to {end_date} |
|
|
- Type of traveler: {traveler_type} |
|
|
- Traveling with: {companion} |
|
|
- Each day should have 3 activities for morning, afternoon, and night. |
|
|
- Provide unique local recommendations based on the traveler's interests. |
|
|
""" |
|
|
|
|
|
response = chatbot(prompt, max_length=1000, do_sample=True) |
|
|
return response[0]['generated_text'] |
|
|
|
|
|
|
|
|
interface = gr.Interface( |
|
|
fn=generate_itinerary, |
|
|
inputs=[ |
|
|
gr.Textbox(label="Destination"), |
|
|
gr.Textbox(label="Start Date (YYYY-MM-DD)"), |
|
|
gr.Textbox(label="End Date (YYYY-MM-DD)"), |
|
|
gr.Dropdown(["Art", "Food", "Nature", "Culture", "Nightlife"], label="Traveler Type"), |
|
|
gr.Dropdown(["Alone", "With Family", "With Friends", "With Partner"], label="Companion Type"), |
|
|
], |
|
|
outputs="text", |
|
|
title="✈️ AI Travel Assistant", |
|
|
description="Enter your details, and let AI plan your perfect trip!" |
|
|
) |
|
|
|
|
|
interface.launch() |
|
|
|
