Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import json
 
 # Load pre-trained model and tokenizer (replace with desired model if needed)
-model_name = "microsoft/DialoGPT-large"
+model_name = "microsoft/DialoGPT-large"  # Replace with your preferred model
 model = AutoModelForCausalLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
@@ -49,10 +49,23 @@ def chat(message, history):
             response += "\nError processing the JSON data."
 
     # Update history with current conversation (optional)
-    history.append([message, response])
+    # history.append([message, response])  # Uncomment if you want conversation history
 
     return response
 
-# Create Gradio interface for the chatbot
-interface = gr.Interface(chat, inputs="textbox", outputs="textbox", catch_exceptions=True)
-interface.launch(share=True)  # Launch the Gradio app and share link
+# Check Gradio version and handle catch_exceptions accordingly
+try:
+    # Option 1: Use catch_exceptions if Gradio supports it
+    interface = gr.Interface(chat, inputs="textbox", outputs="textbox", catch_exceptions=True)
+except TypeError:  # If catch_exceptions is not supported
+    # Option 2: Manual error handling within chat function
+    def chat_with_error_handling(message, history):
+        try:
+            return chat(message, history)
+        except Exception as e:
+            return f"An error occurred: {str(e)}"
+
+    interface = gr.Interface(chat_with_error_handling, inputs="textbox", outputs="textbox")
+
+# Launch the Gradio app and share link
+interface.launch(share=True)