Update app.py

app.py

@@ -25,13 +25,18 @@ transform = transforms.Compose([
 LABELS_URL = "https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json"
 labels = json.loads(requests.get(LABELS_URL).text)
 
-def chat_with_gemini(message, chat_history):
+# Global chat history variable
+chat_history = []
+
+def chat_with_gemini(message):
+    global chat_history
     # Get a response from the language model
     bot_response = llm.predict(message)
     chat_history.append((message, bot_response))
     return chat_history
 
-def analyze_image(image_path, chat_history):
+def analyze_image(image_path):
+    global chat_history
     # Open, preprocess, and classify the image
     image = Image.open(image_path).convert("RGB")
     image_tensor = transform(image).unsqueeze(0)
@@ -58,19 +63,12 @@ with gr.Blocks() as demo:
     send_btn = gr.Button("Send")
     img_upload = gr.Image(type="filepath", label="Upload an image for analysis")
 
-    # Local chat history state
-    chat_history = []
-
     # Define interactions
     def handle_text_message(message):
-        nonlocal chat_history
-        chat_history = chat_with_gemini(message, chat_history)
-        return chat_history
+        return chat_with_gemini(message)
 
     def handle_image_upload(image_path):
-        nonlocal chat_history
-        chat_history = analyze_image(image_path, chat_history)
-        return chat_history
+        return analyze_image(image_path)
 
     # Set up Gradio components with Enter key for sending
     msg.submit(handle_text_message, msg, chatbot)
@@ -106,4 +104,5 @@ with gr.Blocks() as demo:
     </style>
     """)
 
-iface.launch(debug=True)
+# Launch for Hugging Face Spaces
+demo.launch()