solve const issue

app.py

@@ -1,9 +1,31 @@
+# First import our patch before any gradio imports
+try:
+    from patch_gradio import patched_get_type
+except ImportError:
+    # Create patch on the fly if file doesn't exist
+    from types import MethodType
+    import gradio_client.utils
+    
+    # Store the original function
+    original_get_type = gradio_client.utils.get_type
+    
+    # Define the patched function
+    def patched_get_type(schema):
+        # Check if schema is a dict before using 'in' operator
+        if not isinstance(schema, dict):
+            return str(schema)  # Convert to string as fallback
+        
+        # Original logic continues
+        return original_get_type(schema)
+    
+    # Apply the patch
+    gradio_client.utils.get_type = patched_get_type
+
+# Now import the rest of the packages
 import gradio as gr
 from pathlib import Path
 from tempfile import mkdtemp
 
-import patch_gradio
-
 # LangChain & Embedding/LLM
 from langchain_community.vectorstores import FAISS
 from langchain_huggingface.embeddings import HuggingFaceEmbeddings
@@ -198,6 +220,25 @@ def chat_with_doc(query):
     
     return answer, source_text
 
+# Graceful error handling wrapper for Gradio functions
+def safe_process_file(file, extraction_method):
+    try:
+        return process_file(file, extraction_method)
+    except Exception as e:
+        import traceback
+        error_trace = traceback.format_exc()
+        print(f"Error processing file: {str(e)}\n{error_trace}")
+        return f"Error processing file: {str(e)}"
+
+def safe_chat_with_doc(query):
+    try:
+        return chat_with_doc(query)
+    except Exception as e:
+        import traceback
+        error_trace = traceback.format_exc()
+        print(f"Error processing query: {str(e)}\n{error_trace}")
+        return f"Error processing query: {str(e)}", ""
+
 # Gradio Interface
 with gr.Blocks() as app:
     gr.Markdown("# Document Q&A System")
@@ -214,7 +255,7 @@ with gr.Blocks() as app:
         file_output = gr.Textbox(label="Processing Status", interactive=False)
         
         process_btn.click(
-            fn=process_file,
+            fn=safe_process_file,  # Use the wrapped safe function
             inputs=[file_input, extraction_method],
             outputs=file_output
         )
@@ -226,13 +267,13 @@ with gr.Blocks() as app:
         sources_output = gr.Textbox(label="Sources", interactive=False)
         
         ask_btn.click(
-            fn=chat_with_doc,
+            fn=safe_chat_with_doc,  # Use the wrapped safe function
             inputs=chat_input,
             outputs=[chat_output, sources_output]
         )
         
         chat_input.submit(
-            fn=chat_with_doc,  
+            fn=safe_chat_with_doc,  # Use the wrapped safe function
             inputs=chat_input,
             outputs=[chat_output, sources_output]
         )

patch_gradio.py

@@ -1,17 +1,18 @@
 # Create a file called patch_gradio.py
+from types import MethodType
 import gradio_client.utils
 
 # Store the original function
 original_get_type = gradio_client.utils.get_type
 
 # Define the patched function
-def patched_get_type(schema):
-    # Add type checking before using 'in' operator
+def patched_get_type(self, schema):
+    # Check if schema is a dict before using 'in' operator
     if not isinstance(schema, dict):
         return str(schema)  # Convert to string as fallback
     
-    # Original function logic
-    return original_get_type(schema)
+    # Original logic continues
+    return original_get_type(self, schema)
 
 # Apply the patch
 gradio_client.utils.get_type = patched_get_type