Update app.py

app.py

@@ -26,16 +26,19 @@ def chat_with_ai(message, history):
 
         # Check if result is valid and return the response in the correct format
         if result:
-            return [{"role": "assistant", "content": result}]
+            # Append new message to history
+            history.append((message, result))
+            return history, history  # Return updated history for display
         else:
-            return [{"role": "assistant", "content": "Sorry, I couldn't generate a response."}]
-    
+            return history, history
+
     except Exception as e:
         print(f"Error during model prediction: {e}")
-        return [{"role": "assistant", "content": "There was an error processing your message."}]
+        return history, [{"role": "assistant", "content": "There was an error processing your message."}]
 
 # Gradio Chat Interface
-demo = gr.ChatInterface(fn=chat_with_ai)
+demo = gr.ChatInterface(fn=chat_with_ai, allow_screenshot=False)
 
 if __name__ == "__main__":
     demo.launch()
+