Spaces:
Sleeping
Sleeping
| from travel_agent import travel_agent | |
| import gradio as gr | |
| import os | |
| from langchain_google_genai import ChatGoogleGenerativeAI | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| GOOGLE_API_KEY=os.getenv('google_api_key') | |
| GEMINI_MODEL='gemini-2.0-flash' | |
| llm = ChatGoogleGenerativeAI(google_api_key=GOOGLE_API_KEY, model=GEMINI_MODEL, temperature=0.3) | |
| travel_assistant=travel_agent(llm) | |
| def chat(input, history): | |
| response=travel_assistant.chat(input) | |
| return response | |
| #initializing the agent | |
| with gr.Blocks(title="Travel Assistant") as app: | |
| gr.Markdown("# Travel Assistant") | |
| gr.Markdown("Upload an image of your schedule, then ask questions about it.") | |
| with gr.Row(): | |
| with gr.Sidebar(): | |
| with gr.Column(scale=1): | |
| image_input = gr.Image(label="Upload Schedule Image", type="pil") | |
| upload_button = gr.Button("Process Schedule") | |
| with gr.Column(scale=2): | |
| gr.ChatInterface(chat, type="messages", autofocus=False, ) | |
| upload_button.click( | |
| fn=travel_assistant.image_processing, | |
| inputs=image_input | |
| ) | |
| # message.submit( | |
| # fn=chat, | |
| # inputs=[message, chatbot], | |
| # outputs=[chatbot] | |
| # ) | |
| gr.Markdown("## Example Questions") | |
| gr.Markdown(""" | |
| - "Show me my full schedule" | |
| - "What can you do?" | |
| - "Do I have anything at 2:00 PM?" | |
| - "Find a nice vegetarian restaurant in (location)" | |
| - "edit my schedule to change the time of (event)" | |
| - "When am I free on Tuesday?" | |
| """) | |
| # Launch the app | |
| if __name__ == "__main__": | |
| app.launch() |