Update app.py

app.py

@@ -2,15 +2,33 @@ 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
+# Fix for SessionMiddleware error in HF Spaces
+def configure_gradio_session():
+    """Configure session middleware for Gradio in HF Spaces"""
+    try:
+        from starlette.middleware.sessions import SessionMiddleware
+        import gradio.routes
+        
+        # Store original create_app function
+        if hasattr(gradio.routes, 'App') and hasattr(gradio.routes.App, 'create_app'):
+            original_create_app = gradio.routes.App.create_app
+            
+            def patched_create_app(self, *args, **kwargs):
+                app = original_create_app(self, *args, **kwargs)
+                # Add session middleware
+                app.add_middleware(
+                    SessionMiddleware,
+                    secret_key=os.getenv("HF_SESSION_SECRET", secrets.token_hex(32))
+                )
+                return app
+            
+            gradio.routes.App.create_app = patched_create_app
+            
+    except (ImportError, AttributeError) as e:
+        print(f"Warning: Could not configure session middleware: {e}")
 
-# Configure session middleware for Gradio
-if hasattr(gr, 'configure_session'):
-    gr.configure_session(secret_key=secrets.token_hex(32))
+# Call the configuration function
+configure_gradio_session()
 
 # Load math datasets for sample problems
 ds = load_dataset("HuggingFaceH4/ultrachat_200k", streaming=True)