Update app.py

app.py

@@ -1,86 +1,100 @@
-import os
-from dotenv import load_dotenv
+# Importing libraries
 import streamlit as st
-from crewai import Agent, Task, Crew, Process
 import google.generativeai as genai
 
-# === Load environment variables ===
-load_dotenv()  # loads API_KEY from .env
-
-# === API Key Config ===
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
-os.environ["GEMINI_API_KEY"] = GEMINI_API_KEY  # Set for potential other uses
-
-# Configure Gemini
+# Configuring Gemini API Key
+GEMINI_API_KEY = "AIzaSyCdMDLOh9jv4H7Mj-4Qtq8f_SAuuPejAHE"  # Replace with your actual API key
 genai.configure(api_key=GEMINI_API_KEY)
 
-# === Streamlit UI Config ===
-st.set_page_config(
-    page_title="☕ Java Coding Assistant",
-    page_icon="☕",
-    layout="wide"
-)
+# Function to query Gemini API to generate response
+def generate_response(prompt):
+    """Uses Gemini API to generate AI-powered responses."""
+    model = genai.GenerativeModel("gemini-2.0-flash")
+    response = model.generate_content(prompt)
+    return response.text
 
-# Initialize chat history
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+# Function to refine and clarify user inputs
+def refine_user_inputs(inputs):
+    """Checks for missing details and asks clarifying questions."""
+    missing_fields = []
+    
+    # Identify missing inputs
+    for key, value in inputs.items():
+        if not value or value == "Select an option":
+            missing_fields.append(key)
 
-# Java Coding Assistant function
-def java_coding_assistant(query: str, temperature: float = 0.7) -> str:
-    try:
-        # Initialize Gemini model
-        model = genai.GenerativeModel('gemini-1.5-pro')
-        
-        # Create prompt with the query
+    if missing_fields:
+        # Generate a dynamic clarification prompt
+        missing_fields_str = ", ".join(missing_fields)
         prompt = f"""
-        You are a Senior Java Full Stack Developer with 10+ years of experience.
-        Provide a comprehensive solution for this Java-related query:
-        {query}
-        
-        Your response should include:
-        1. Problem analysis
-        2. Complete Java code solution with proper syntax
-        3. Explanation of key concepts
-        4. Best practices
-        5. Potential edge cases
-        
-        Format your response with:
-        - Problem breakdown
-        - Complete Java code with syntax highlighting (use ```java markers)
-        - Detailed explanation
-        - Best practices
-        - Runtime complexity analysis (if applicable)
+        The user has provided incomplete travel details. They have not specified the following: {missing_fields_str}.
+        Ask clear, user-friendly follow-up questions to fill in these gaps.
         """
-        
-        # Generate response
-        response = model.generate_content(prompt)
-        
-        # Get the text from the response
-        result = response.text
-        
-        # Replace java fences for Streamlit if needed
-        if "```java" in result:
-            result = result.replace("```java", "```python")
-        return result
+        return generate_response(prompt)
+    
+    return "Thank you for providing all details! Proceeding with itinerary generation for required no.of days."
+
+# Function to generate a detailed travel itinerary
+def generate_itinerary(user_inputs):
+    """Creates a structured day-by-day itinerary based on user preferences."""
+    prompt = f"""
+    Generate a **{user_inputs.get('trip_duration')}-day travel itinerary** for {user_inputs.get('destination')}.
+    Customize it based on:
+    - Budget: {user_inputs.get('budget')}
+    - Interests: {user_inputs.get('preferences')}
+    - Purpose: {user_inputs.get('purpose')}
+    - Accommodation: {user_inputs.get('accommodation')}
+    - Transport: {user_inputs.get('transport')}
+    - Dietary Needs: {user_inputs.get('dietary')}
+
+    Ensure the itinerary includes:
+    - Morning, Afternoon, and Evening Activities
+    - Top attractions & restaurants
+    - Relaxation time & hidden gems
+    - Logical time slots for activities
+    """
+    return generate_response(prompt)
 
-    except Exception as e:
-        return f"❌ Error: {e}"
+# 🔹 Streamlit UI
+st.title("AI-Powered Traveler Guide")
+st.write("Plan your trip with a personalized AI-generated itinerary!")
 
-# === Streamlit interface ===
-st.title("☕ Java Coding Assistant")
-st.caption("Ask me anything about Java backend, Android development, or coding problems")
+# User input fields
+destination = st.text_input("Where are you traveling to?")
+starting_location = st.text_input("Where are you traveling from?")
+budget = st.selectbox("Budget:", ["Select an option", "Low", "Mid-range", "Luxury"])
+trip_duration = st.number_input("Trip Duration (days):", min_value=1, max_value=30, step=1)
+travel_dates = st.text_input("Provide me Travel Dates (e.g., Jan 10 - Jan 17):")
+purpose = st.selectbox("Trip Purpose:", ["Select an option", "Vacation", "Business", "Adventure", "Honeymoon", "Cultural"])
+preferences = st.multiselect("What interests you?", ["Nature", "Culture", "Food", "Shopping", "Adventure", "History"])
+accommodation = st.selectbox("Accommodation Preference:", ["Select an option", "Luxury Hotel", "Budget Hotel", "Hostel", "Airbnb"])
+transport = st.selectbox("Preferred Transport:", ["Select an option", "Public Transport", "Car Rental", "Walking", "Private Tours"])
+dietary = st.selectbox("Dietary Preferences:", ["Select an option", "No Preference", "Vegetarian", "Vegan", "Non-Vegetarian", "Halal"])
 
-# Render chat history
-for msg in st.session_state.messages:
-    with st.chat_message(msg["role"]):
-        st.markdown(msg["content"])
+# Button to generate itinerary
+if st.button("Generate Itinerary"):
+    user_inputs = {
+        "destination": destination,
+        "starting_location": starting_location,
+        "budget": budget,
+        "trip_duration": trip_duration,
+        "travel_dates": travel_dates,
+        "purpose": purpose,
+        "preferences": ", ".join(preferences) if preferences else "Not specified",
+        "accommodation": accommodation,
+        "transport": transport,
+        "dietary": dietary,
+    }
+    
+    # Step 1: Check and clarify missing inputs
+    st.write("Checking the provided travel details...")
+    refinement_message = refine_user_inputs(user_inputs)
+    st.write(refinement_message)
 
-# Chat input
-if user_input := st.chat_input("Ask your Java question (e.g., 'How to implement REST API in Spring Boot?'):"):
-    st.session_state.messages.append({"role": "user", "content": user_input})
-    with st.chat_message("user"):
-        st.markdown(user_input)
-    with st.chat_message("assistant"):
-        answer = java_coding_assistant(user_input)
-        st.markdown(answer, unsafe_allow_html=True)
-    st.session_state.messages.append({"role": "assistant", "content": answer})
+    if "Thank you for providing all details!" in refinement_message:
+        # Step 2: Generate Itinerary
+        st.write("Creating your AI-powered travel itinerary...")
+        itinerary = generate_itinerary(user_inputs)
+        st.write(itinerary)
+    else:
+        st.warning("Some details are missing, please provide the missing details before generating the itinerary.")