Update main.py

main.py

@@ -1,8 +1,8 @@
-import streamlit as st
+import gradio as gr
 from langchain_experimental.llms.ollama_functions import OllamaFunctions
 
+# Initialize the Ollama model
 model = OllamaFunctions(model="gemma:7b")
-
 model = model.bind(
     functions=[
         {
@@ -13,7 +13,7 @@ model = model.bind(
                 "properties": {
                     "location": {
                         "type": "string",
-                        "description": "The city and state, " "e.g. San Francisco, CA",
+                        "description": "The city and state, e.g., San Francisco, CA",
                     },
                     "unit": {
                         "type": "string",
@@ -27,5 +27,19 @@ model = model.bind(
     function_call={"name": "get_current_weather"},
 )
 
-user_input = st.text_input("Enter your question:")
-model.invoke(user_input)
+# Function to be called by the Gradio interface
+def get_weather(user_input):
+    result = model.invoke(user_input)
+    return result
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=get_weather, 
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter Location and Unit (e.g., 'San Francisco, CA', 'celsius')"), 
+    outputs="text",
+    title="Weather Information",
+    description="Enter a location to get the current weather."
+)
+
+# Launch the application
+iface.launch()