Update app.py

app.py

@@ -3,6 +3,8 @@ import re
 import tempfile
 import requests
 import gradio as gr
+print(f"Gradio version: {gr.__version__}")
+
 from PyPDF2 import PdfReader
 import logging
 import webbrowser
@@ -627,19 +629,42 @@ with gr.Blocks(css="""
             first_model = list(ctx_size.keys())[0]
             ctx_size = ctx_size[first_model]
             
-            # Update model dropdown based on provider
+            # Prepare dropdown choices based on provider
             if choice == "OpenAI ChatGPT":
-                openai_model.update(choices=list(MODEL_CONTEXT_SIZES["OpenAI ChatGPT"].keys()))
+                model_choices = list(MODEL_CONTEXT_SIZES["OpenAI ChatGPT"].keys())
+                return [
+                    gr.update(visible=False),  # hf_options
+                    gr.update(visible=False),  # groq_options
+                    gr.update(visible=True),   # openai_options
+                    gr.update(value=ctx_size), # context_size
+                    gr.Dropdown(choices=model_choices, value=first_model)  # openai_model
+                ]
             elif choice == "HuggingFace Inference":
-                hf_model.update(choices=list(MODEL_CONTEXT_SIZES["HuggingFace Inference"].keys()))
+                model_choices = list(model_registry.hf_models.keys())
+                return [
+                    gr.update(visible=True),   # hf_options
+                    gr.update(visible=False),  # groq_options
+                    gr.update(visible=False),  # openai_options
+                    gr.update(value=ctx_size), # context_size
+                    gr.Dropdown(choices=model_choices, value="Phi-3 Mini 4K")  # openai_model (not used)
+                ]
             elif choice == "Groq API":
-                groq_model.update(choices=list(MODEL_CONTEXT_SIZES["Groq API"].keys()))
+                model_choices = list(model_registry.groq_models.keys())
+                return [
+                    gr.update(visible=False),  # hf_options
+                    gr.update(visible=True),   # groq_options
+                    gr.update(visible=False),  # openai_options
+                    gr.update(value=ctx_size), # context_size
+                    gr.Dropdown(choices=model_choices, value=model_choices[0] if model_choices else None)  # openai_model (not used)
+                ]
         
+        # Default return for "Clipboard only" or other options
         return [
-            gr.update(visible=choice == "HuggingFace Inference"),
-            gr.update(visible=choice == "Groq API"),
-            gr.update(visible=choice == "OpenAI ChatGPT"),
-            gr.update(value=ctx_size)
+            gr.update(visible=False),  # hf_options
+            gr.update(visible=False),  # groq_options
+            gr.update(visible=False),  # openai_options
+            gr.update(value=4096),    # context_size
+            gr.Dropdown(choices=[])    # openai_model (not used)
         ]
     
     # PDF Processing Handlers
@@ -784,7 +809,8 @@ with gr.Blocks(css="""
             hf_options,
             groq_options,
             openai_options,
-            context_size
+            context_size,
+            openai_model
         ]
     )