Update app.py

app.py

@@ -265,24 +265,23 @@ def send_to_openai(prompt: str, api_key: str) -> str:
         logging.error(f"Error with OpenAI API: {e}")
         return f"Error with OpenAI API: {e}"
 
-def copy_to_clipboard(element_id: str) -> str:
+def copy_text_js(element_id: str) -> str:
     return f"""
         () => {{
             try {{
-                const text = document.querySelector('#{element_id} textarea').value;
+                const elem = document.querySelector('#{element_id} textarea');
+                if (!elem) throw new Error('Element not found');
+                const text = elem.value;
+                if (!text) throw new Error('No text to copy');
                 navigator.clipboard.writeText(text);
                 return "Copied to clipboard!";
             }} catch (e) {{
                 console.error(e);
-                return "Failed to copy to clipboard";
+                return "Failed to copy: " + e.message;
             }}
         }}
     """
 
-def open_chatgpt_old() -> str:
-    webbrowser.open_new_tab('https://chat.openai.com')
-    return "Opening ChatGPT in new tab"
-
 def open_chatgpt() -> str:
     """Open ChatGPT in new browser tab"""
     return """window.open('https://chat.openai.com/', '_blank');"""
@@ -583,38 +582,38 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
     def handle_model_selection(choice):
         """Handle model selection and update UI"""
         ctx_size = get_model_context_size(choice)
-        return {
-            hf_options: gr.update(visible=choice == "HuggingFace Inference"),
-            groq_options: gr.update(visible=choice == "Groq API"),
-            openai_options: gr.update(visible=choice == "OpenAI ChatGPT"),
-            context_size: gr.update(value=ctx_size)
-        }
+        return (
+            gr.update(visible=choice == "HuggingFace Inference"),  # hf_options
+            gr.update(visible=choice == "Groq API"),              # groq_options
+            gr.update(visible=choice == "OpenAI ChatGPT"),        # openai_options
+            gr.update(value=ctx_size)                            # context_size
+        )
     
     # PDF Processing Handlers
     def handle_pdf_process(pdf, fmt, ctx_size):
         """Process PDF and update UI state"""
         if not pdf:
-            return {
-                progress_status: "Please upload a PDF file.",
-                processed_text: "",
-                pdf_content: "",
-                snippets: [],
-                snippet_selector: gr.update(choices=[], value=None),
-                download_files: None
-            }
+            return (
+                "Please upload a PDF file.",  # progress_status
+                "",                          # processed_text
+                "",                          # pdf_content
+                [],                         # snippets
+                gr.update(choices=[], value=None),  # snippet_selector
+                None                        # download_files
+            )
             
         try:
             # Extract and format text
             text = extract_text_from_pdf(pdf.name)
             if text.startswith("Error"):
-                return {
-                    progress_status: text,
-                    processed_text: "",
-                    pdf_content: "",
-                    snippets: [],
-                    snippet_selector: gr.update(choices=[], value=None),
-                    download_files: None
-                }
+                return (
+                    text,
+                    "",
+                    "",
+                    [],
+                    gr.update(choices=[], value=None),
+                    None
+                )
                 
             formatted_text = format_content(text, fmt)
             snippets_list = split_into_snippets(formatted_text, ctx_size)
@@ -624,35 +623,35 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
                 f.write(formatted_text)
                 download_file = f.name
                 
-            return {
-                progress_status: f"PDF processed successfully! Generated {len(snippets_list)} snippets.",
-                processed_text: formatted_text,
-                pdf_content: formatted_text,
-                snippets: snippets_list,
-                snippet_selector: gr.update(choices=update_snippet_choices(snippets_list), value="Snippet 1 of " + str(len(snippets_list))),
-                download_files: [download_file]
-            }
+            return (
+                f"PDF processed successfully! Generated {len(snippets_list)} snippets.",
+                formatted_text,
+                formatted_text,
+                snippets_list,
+                gr.update(choices=update_snippet_choices(snippets_list), value="Snippet 1 of " + str(len(snippets_list))),
+                [download_file]
+            )
             
         except Exception as e:
             error_msg = f"Error processing PDF: {str(e)}"
             logging.error(error_msg)
-            return {
-                progress_status: error_msg,
-                processed_text: "",
-                pdf_content: "",
-                snippets: [],
-                snippet_selector: gr.update(choices=[], value=None),
-                download_files: None
-            }
+            return (
+                error_msg,
+                "",
+                "",
+                [],
+                gr.update(choices=[], value=None),
+                None
+            )
 
     def handle_snippet_selection(choice, snippets_list):
         """Handle snippet selection and update prompt"""
         if not snippets_list:
-            return {
-                progress_status: "No snippets available.",
-                generated_prompt: "",
-                download_files: None
-            }
+            return (
+                "No snippets available.",  # progress_status
+                "",                       # generated_prompt
+                None                      # download_files
+            )
             
         try:
             idx = get_snippet_index(choice)
@@ -662,40 +661,27 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
             with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as f:
                 f.write(selected_snippet)
                 
-            return {
-                progress_status: f"Selected snippet {idx + 1}",
-                generated_prompt: selected_snippet,
-                download_files: [f.name]
-            }
+            return (
+                f"Selected snippet {idx + 1}",
+                selected_snippet,
+                [f.name]
+            )
             
         except Exception as e:
             error_msg = f"Error selecting snippet: {str(e)}"
             logging.error(error_msg)
-            return {
-                progress_status: error_msg,
-                generated_prompt: "",
-                download_files: None
-            }
+            return (
+                error_msg,
+                "",
+                None
+            )
         
     # Copy button handlers
-    def copy_text_js(element_id: str) -> str:
-        return f"""
-            () => {{
-                const text = document.querySelector('#{element_id} textarea').value;
-                navigator.clipboard.writeText(text);
-                return "Copied to clipboard!";
-            }}
-        """
-
     def handle_prompt_generation(snippet_text, template, snippet_choice, snippets_list):
         """Generate prompt from selected snippet"""
         if not snippet_text or not snippets_list:
-            return {
-                progress_status: "No text available for prompt generation.",
-                generated_prompt: "",
-                download_files: None
-            }
-            
+            return "No text available for prompt generation.", "", None
+                
         try:
             idx = get_snippet_index(snippet_choice)
             prompt = generate_prompt(snippets_list[idx], template or "Summarize the following text:")
@@ -704,20 +690,12 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
             with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as f:
                 f.write(prompt)
                 
-            return {
-                progress_status: "Prompt generated successfully!",
-                generated_prompt: prompt,
-                download_files: [f.name]
-            }
+            return "Prompt generated successfully!", prompt, [f.name]
             
         except Exception as e:
             error_msg = f"Error generating prompt: {str(e)}"
             logging.error(error_msg)
-            return {
-                progress_status: error_msg,
-                generated_prompt: "",
-                download_files: None
-            }
+            return error_msg, "", None
 
     def handle_copy_action(text):
         """Handle copy to clipboard action"""
@@ -743,34 +721,34 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
     generate_prompt_btn.click(
         handle_prompt_generation,
         inputs=[generated_prompt, custom_prompt, snippet_selector, snippets],
-        outputs={
-            progress_status: progress_status,
-            generated_prompt: generated_prompt,
-            download_files: download_files
-        }
+        outputs=[
+            progress_status,
+            generated_prompt,
+            download_files
+        ]
     )
 
     # Snippet handling
     snippet_selector.change(
         handle_snippet_selection,
         inputs=[snippet_selector, snippets],
-        outputs={
-            progress_status: progress_status,
-            generated_prompt: generated_prompt,
-            download_files: download_files
-        }
+        outputs=[
+            progress_status,
+            generated_prompt,
+            download_files
+        ]
     )
 
     # Model selection
     model_choice.change(
         handle_model_selection,
         inputs=[model_choice],
-        outputs={
-            hf_options: hf_options,
-            groq_options: groq_options,
-            openai_options: openai_options,
-            context_size: context_size
-        }
+        outputs=[
+            hf_options,
+            groq_options,
+            openai_options,
+            context_size
+        ]
     )
 
     hf_model.change(
@@ -786,19 +764,18 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
     )
 
     # Context size buttons
-    """
     for size_name, size_value in CONTEXT_SIZES.items():
-        gr.Button(size_name, size="sm").click(
-            update_context_size,
-            inputs=[],
-            outputs=context_size
-        ).success(
-            lambda s=size_value: int(s),
+        gr.Button(
+            size_name, 
+            size="sm",
+            scale=1
+        ).click(
+            lambda v=size_value: gr.update(value=v),
             None,
             context_size
-        ) """
+        )
 
-    # Download handlers
+    # Download handlers (simplified)
     for btn, content in [
         (download_full_text, pdf_content),
         (download_snippet, generated_prompt),
@@ -808,7 +785,7 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
         btn.click(
             lambda x: [x] if x else None,
             inputs=[content],
-            outputs=download_files
+            outputs=[download_files]
         )
 
     # Copy button handlers