Update app.py

app.py

@@ -2,6 +2,15 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 from datasets import load_dataset
 import random
+import secrets
+
+# Add SessionMiddleware configuration
+import gradio as gr
+from starlette.middleware.sessions import SessionMiddleware
+
+# Configure session middleware for Gradio
+if hasattr(gr, 'configure_session'):
+    gr.configure_session(secret_key=secrets.token_hex(32))
 
 # Load math datasets for sample problems
 ds = load_dataset("HuggingFaceH4/ultrachat_200k", streaming=True)