Update app.py

app.py

@@ -9,6 +9,7 @@ import webbrowser
 from huggingface_hub import InferenceClient
 from typing import Dict, List, Optional, Tuple
 import time
+from groq import Groq  # Import the Groq client
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -527,15 +528,16 @@ with gr.Blocks(css="""
                     )
                
                with gr.Column(visible=False) as groq_options:
-                   groq_model = gr.Dropdown(
-                       choices=list(MODEL_CONTEXT_SIZES["Groq API"].keys()),
-                       value="mixtral-8x7b-32768",
-                       label="Groq Model"
-                   )
-                   groq_api_key = gr.Textbox(
-                       label="🔑 Groq API Key",
-                       type="password"
-                   )
+                    groq_model = gr.Dropdown(
+                        choices=list(model_registry.groq_models.keys()),  # Use model_registry.groq_models
+                        value=list(model_registry.groq_models.keys())[0] if model_registry.groq_models else None, # Set a default value if available
+                        label="Groq Model"
+                    )
+                    groq_api_key = gr.Textbox(
+                        label="🔑 Groq API Key",
+                        type="password"
+                    )
+                    groq_refresh_btn = gr.Button("🔄 Refresh Groq Models")  # Add refresh button
                     
                send_to_model_btn = gr.Button("🚀 Send to Model", variant="primary")
                open_chatgpt_button = gr.Button("🌐 Open ChatGPT")
@@ -808,22 +810,16 @@ with gr.Blocks(css="""
     )
 
     # Download handlers
-    for btn, elem_id in [
-        (copy_prompt_button, "generated_prompt"),
-        (copy_summary_button, "summary_output")
-    ]:
-        btn.click(
-            fn=None,
-            _js=f"""
-            () => {{
-                const el = document.getElementById('{elem_id}');
-                if (!el) return 'Element not found';
-                navigator.clipboard.writeText(el.value);
-                return 'Copied to clipboard!';
-            }}
-            """,
-            outputs=progress_status
-        )
+    copy_prompt_button.click(
+        fn=None,
+        _js=copy_text_js("generated_prompt"), # Use the helper function
+        outputs=progress_status
+    )
+    copy_summary_button.click(
+        fn=None,
+        _js=copy_text_js("summary_output"), # Use the helper function
+        outputs=progress_status
+    )
 
     def download_file(content: str, prefix: str) -> List[str]:
         if not content: