Update app.py

app.py

@@ -13,19 +13,9 @@ def chat(message, history, img=None):
   genai.configure(api_key='AIzaSyCMBk81YmILNTok8hd6tYtJaevp1qbl6I0')  # Replace with your actual API key
 
   text_model = genai.GenerativeModel('gemini-pro')
-  vision_model = genai.GenerativeModel('gemini-pro-vision')
+  #vision_model = genai.GenerativeModel('gemini-pro-vision')
 
   try:
-    if img is not None:
-      # Process image with vision model
-      image_response = vision_model.generate_content(img)
-      image_response.resolve()  # Wait for response completion
-      image_text = to_markdown(image_response.text)
-
-      # Combine image and text for unified response
-      prompt = f"{message}\n{image_text}"
-      response = text_model.generate_content(prompt, stream=True)
-    else:
       # Use only text model
       response = text_model.generate_content(message, stream=True)
 
@@ -40,8 +30,7 @@ chat_interface = gr.ChatInterface(
   fn=chat,
   title="Gemini Chat",
   description="Chat with an AI assistant powered by Gemini",
-  theme="soft",
-  inputs=[gr.Textbox(lines=1, label="Enter your message"), gr.Image()],
+  theme="soft"
 )
 
 chat_interface.launch()