File size: 1,461 Bytes
cb2beda |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import gradio as gr
from transformers import pipeline
# Load Mistral-7B-Instruct AI model
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']
# Creating a simple user interface
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()
|