Update app.py

app.py

@@ -227,14 +227,15 @@ def extract_with_gemini(processed_data: Dict[str, Any], api_key: str, model_name
         }
 
 
-def process_documents(files, model_choice):
+def process_documents(files):
     """Main Gradio processing function"""
     
     if not files or len(files) == 0:
         return "❌ Error: Please upload at least one file", "{}", "No files provided"
     
-    # Use the hardcoded API key
+    # Use the hardcoded API key and default model
     api_key = GEMINI_API_KEY
+    model_choice = "gemini-2.0-flash-exp"
     
     if not api_key or api_key.strip() == "":
         return "❌ Error: API key not configured in code", "{}", "API key missing"
@@ -293,20 +294,6 @@ def create_interface():
         
         with gr.Row():
             with gr.Column(scale=2):
-                api_key_input = gr.Textbox(
-                    label="🔑 Gemini API Key",
-                    placeholder="Enter your Google Gemini API key (AIza...)",
-                    type="password",
-                    info="Get your key from https://aistudio.google.com/apikey"
-                )
-                
-                model_choice = gr.Dropdown(
-                    choices=["gemini-2.0-flash-exp", "gemini-1.5-pro", "gemini-1.5-flash"],
-                    value="gemini-2.0-flash-exp",
-                    label="Model Selection",
-                    info="Latest model recommended for best results"
-                )
-                
                 file_input = gr.File(
                     label="📎 Upload Documents",
                     file_count="multiple",
@@ -345,18 +332,18 @@ def create_interface():
         # Button action
         submit_btn.click(
             fn=process_documents,
-            inputs=[file_input, api_key_input, model_choice],
+            inputs=[file_input],
             outputs=[status_output, json_output, display_output]
         )
         
         # Examples
-        # gr.Examples(
-        #     examples=[
-        #         [["example1.pdf"], "your-api-key-here"],
-        #     ],
-        #     inputs=[file_input, api_key_input],
-        #     label="Example Usage"
-        # )
+        gr.Examples(
+            examples=[
+                [["example1.pdf"]],
+            ],
+            inputs=[file_input],
+            label="Example Usage"
+        )
     
     return demo