Upload Gradio_Related.py

Fix for fstring + setting Mistral to default API wtih api key set as env secret since free requests for 15 days.

App_Function_Libraries/Gradio_Related.py

@@ -669,8 +669,11 @@ def create_video_transcription_tab():
                 api_name_input = gr.Dropdown(
                     choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
                              "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                    value=None, label="API Name (Mandatory)")
-                api_key_input = gr.Textbox(label="API Key (Mandatory)", placeholder="Enter your API key here", type="password")
+                    value="Mistral",
+                    label="API Name (Mandatory)")
+                api_key_input = gr.Textbox(label="API Key (Mandatory)", placeholder="Enter your API key here",
+                                           type="password",
+                                           value=f"{MISTRAL_TOKEN}")
                 keywords_input = gr.Textbox(label="Keywords", placeholder="Enter keywords here (comma-separated)",
                                             value="default,no_keyword_set")
                 batch_size_input = gr.Slider(minimum=1, maximum=10, value=1, step=1,
@@ -1316,10 +1319,12 @@ def create_audio_processing_tab():
                 api_name_input = gr.Dropdown(
                     choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
                              "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                    value=None,
+                    value="Mistral",
                     label="API for Summarization (Optional)"
                 )
-                api_key_input = gr.Textbox(label="API Key (if required)", placeholder="Enter your API key here", type="password")
+                api_key_input = gr.Textbox(label="API Key (if required)", placeholder="Enter your API key here",
+                                           type="password",
+                                           value=f"{MISTRAL_TOKEN}")
                 custom_keywords_input = gr.Textbox(label="Custom Keywords", placeholder="Enter custom keywords, comma-separated")
                 keep_original_input = gr.Checkbox(label="Keep original audio file", value=False)
 
@@ -1424,10 +1429,12 @@ def create_podcast_tab():
                 podcast_api_name_input = gr.Dropdown(
                     choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter", "Llama.cpp",
                              "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                    value=None,
+                    value="Mistral",
                     label="API Name for Summarization (Optional)"
                 )
-                podcast_api_key_input = gr.Textbox(label="API Key (if required)", type="password")
+                podcast_api_key_input = gr.Textbox(label="API Key (if required)",
+                                                   type="password",
+                                                   value=f"{MISTRAL_TOKEN}")
                 podcast_whisper_model_input = gr.Dropdown(choices=whisper_models, value="medium", label="Whisper Model")
 
                 keep_original_input = gr.Checkbox(label="Keep original audio file", value=False)
@@ -1545,9 +1552,13 @@ def create_website_scraping_tab():
 
                 api_name_input = gr.Dropdown(
                     choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
-                             "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"], value=None, label="API Name (Mandatory for Summarization)")
+                             "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
+                    value="Mistral",
+                    label="API Name (Mandatory for Summarization)")
                 api_key_input = gr.Textbox(label="API Key (Mandatory if API Name is specified)",
-                                           placeholder="Enter your API key here; Ignore if using Local API or Built-in API", type="password")
+                                           placeholder="Enter your API key here; Ignore if using Local API or Built-in API",
+                                           type="password",
+                                           value=f"{MISTRAL_TOKEN}")
                 keywords_input = gr.Textbox(label="Keywords", placeholder="Enter keywords here (comma-separated)",
                                             value="default,no_keyword_set", visible=True)
 
@@ -1655,8 +1666,12 @@ def create_resummary_tab():
                     api_name_input = gr.Dropdown(
                         choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
                              "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                        value="Local-LLM", label="API Name")
-                    api_key_input = gr.Textbox(label="API Key", placeholder="Enter your API key here", type="password")
+                        value="Mistral",
+                        label="API Name")
+                    api_key_input = gr.Textbox(label="API Key",
+                                               placeholder="Enter your API key here",
+                                               type="password",
+                                               value=f"{MISTRAL_TOKEN}")
 
                 chunking_options_checkbox = gr.Checkbox(label="Use Chunking", value=False)
                 with gr.Row(visible=False) as chunking_options_box:
@@ -2853,9 +2868,13 @@ def create_chat_interface():
                 with gr.Row():
                     load_conversations_btn = gr.Button("Load Selected Conversation")
 
-                api_endpoint = gr.Dropdown(label="Select API Endpoint", choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
-                             "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"])
-                api_key = gr.Textbox(label="API Key (if required)", type="password")
+                api_endpoint = gr.Dropdown(label="Select API Endpoint",
+                                           choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
+                             "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
+                                           value="Mistral")
+                api_key = gr.Textbox(label="API Key (if required)",
+                                     type="password",
+                                     value=f"{MISTRAL_TOKEN}")
                 custom_prompt_checkbox = gr.Checkbox(label="Use a Custom Prompt",
                                                      value=False,
                                                      visible=True)
@@ -3062,8 +3081,11 @@ def create_chat_interface_editable():
 
                 api_endpoint = gr.Dropdown(label="Select API Endpoint",
                                            choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
-                             "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"])
-                api_key = gr.Textbox(label="API Key (if required)", type="password")
+                             "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
+                                           value="Mistral")
+                api_key = gr.Textbox(label="API Key (if required)",
+                                     type="password",
+                                     value=f"{MISTRAL_TOKEN}")
                 # preset_prompt_checkbox = gr.Checkbox(label="Use a pre-set Prompt",
                 #                                      value=False,
                 #                                      visible=True)
@@ -3261,8 +3283,12 @@ def create_chat_interface_stacked():
                     search_conversations_btn = gr.Button("Search Conversations")
                     load_conversations_btn = gr.Button("Load Selected Conversation")
             with gr.Column():
-                api_endpoint = gr.Dropdown(label="Select API Endpoint", choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "OpenRouter", "Mistral", "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"])
-                api_key = gr.Textbox(label="API Key (if required)", type="password")
+                api_endpoint = gr.Dropdown(label="Select API Endpoint",
+                                           choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "OpenRouter", "Mistral", "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
+                                           value="Mistral")
+                api_key = gr.Textbox(label="API Key (if required)",
+                                     type="password",
+                                     value=f"{MISTRAL_TOKEN}")
                 preset_prompt = gr.Dropdown(label="Select Preset Prompt",
                                             choices=load_preset_prompts(),
                                             visible=True)
@@ -3432,8 +3458,11 @@ def create_chat_interface_multi_api():
                     api_endpoint = gr.Dropdown(label=f"API Endpoint {i + 1}",
                                                choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq",
                                                         "DeepSeek", "Mistral", "OpenRouter", "Llama.cpp", "Kobold", "Ooba",
-                                                        "Tabbyapi", "VLLM", "HuggingFace"])
-                    api_key = gr.Textbox(label=f"API Key {i + 1} (if required)", type="password")
+                                                        "Tabbyapi", "VLLM", "HuggingFace"],
+                                               value="Mistral")
+                    api_key = gr.Textbox(label=f"API Key {i + 1} (if required)",
+                                         type="password",
+                                         value=f"{MISTRAL_TOKEN}")
                     temperature = gr.Slider(label=f"Temperature {i + 1}", minimum=0.0, maximum=1.0, step=0.1, value=0.7)
                     chatbot = gr.Chatbot(height=800, elem_classes="chat-window")
                     chatbots.append(chatbot)
@@ -3543,8 +3572,9 @@ def create_chat_interface_four():
                         api_endpoint = gr.Dropdown(label=f"API Endpoint {i + 1}",
                                                    choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq",
                                                             "DeepSeek", "Mistral", "OpenRouter", "Llama.cpp", "Kobold", "Ooba",
-                                                            "Tabbyapi", "VLLM", "HuggingFace"])
-                        api_key = gr.Textbox(label=f"API Key {i + 1} (if required)", type="password")
+                                                            "Tabbyapi", "VLLM", "HuggingFace"],
+                                                   value="Mistral")
+                        api_key = gr.Textbox(label=f"API Key {i + 1} (if required)", type="password", value=f"{MISTRAL_TOKEN}")
                         temperature = gr.Slider(label=f"Temperature {i + 1}", minimum=0.0, maximum=1.0, step=0.1, value=0.7)
                         chatbot = gr.Chatbot(height=400, elem_classes="chat-window")
                         msg = gr.Textbox(label=f"Enter your message for Chat {i + 1}")
@@ -3629,9 +3659,13 @@ def create_chat_interface_vertical():
                     use_prompt = gr.Checkbox(label="Use Prompt")
                     save_conversation = gr.Checkbox(label="Save Conversation", value=False)
                 with gr.Row():
-                    api_endpoint = gr.Dropdown(label="Select API Endpoint", choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter", "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"])
+                    api_endpoint = gr.Dropdown(label="Select API Endpoint",
+                                               choices=["Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter", "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
+                                               value="Mistral")
                 with gr.Row():
-                    api_key = gr.Textbox(label="API Key (if required)", type="password")
+                    api_key = gr.Textbox(label="API Key (if required)",
+                                         type="password",
+                                         value=f"{MISTRAL_TOKEN}")
                 with gr.Row():
                     temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, step=0.1, value=0.7)
                 with gr.Row():
@@ -4325,9 +4359,12 @@ def create_import_item_tab():
             api_name_input = gr.Dropdown(
                 choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
                          "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                label="API for Auto-summarization"
+                label="API for Auto-summarization",
+                value="Mistral"
             )
-            api_key_input = gr.Textbox(label="API Key", type="password")
+            api_key_input = gr.Textbox(label="API Key",
+                                       type="password",
+                                       value=f"{MISTRAL_TOKEN}")
         with gr.Row():
             import_button = gr.Button("Import Data")
         with gr.Row():
@@ -4659,9 +4696,12 @@ def create_import_book_tab():
                 api_name_input = gr.Dropdown(
                     choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
                              "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                    label="API for Auto-summarization"
+                    label="API for Auto-summarization",
+                    value="Mistral"
                 )
-                api_key_input = gr.Textbox(label="API Key", type="password")
+                api_key_input = gr.Textbox(label="API Key",
+                                           type="password",
+                                           value=f"{MISTRAL_TOKEN}")
                 import_button = gr.Button("Import eBook")
             with gr.Column():
                 with gr.Row():
@@ -4761,6 +4801,47 @@ def export_items_by_keyword(keyword: str) -> str:
         return None
 
 
+def export_selected_items(selected_items: List[Dict]) -> Tuple[Optional[str], str]:
+    try:
+        logger.debug(f"Received selected_items: {selected_items}")
+        if not selected_items:
+            logger.warning("No items selected for export")
+            return None, "No items selected for export"
+
+        markdown_content = "# Selected Items\n\n"
+        for item in selected_items:
+            logger.debug(f"Processing item: {item}")
+            try:
+                # Check if 'value' is a string (JSON) or already a dictionary
+                if isinstance(item, str):
+                    item_data = json.loads(item)
+                elif isinstance(item, dict) and 'value' in item:
+                    item_data = item['value'] if isinstance(item['value'], dict) else json.loads(item['value'])
+                else:
+                    item_data = item
+
+                logger.debug(f"Item data after processing: {item_data}")
+
+                if 'id' not in item_data:
+                    logger.error(f"'id' not found in item data: {item_data}")
+                    continue
+
+                content, prompt, summary = fetch_item_details(item_data['id'])
+                markdown_content += f"## {item_data.get('title', 'Item {}'.format(item_data['id']))}\n\n### Prompt\n{prompt}\n\n### Summary\n{summary}\n\n### Content\n{content}\n\n---\n\n"
+            except Exception as e:
+                logger.error(f"Error processing item {item}: {str(e)}")
+                markdown_content += f"## Error\n\nUnable to process this item.\n\n---\n\n"
+
+        filename = "export_selected_items.md"
+        with open(filename, "w", encoding='utf-8') as f:
+            f.write(markdown_content)
+
+        logger.info(f"Successfully exported {len(selected_items)} selected items to {filename}")
+        return filename, f"Successfully exported {len(selected_items)} items to {filename}"
+    except Exception as e:
+        error_message = f"Error exporting selected items: {str(e)}"
+        logger.error(error_message)
+        return None, error_message
 
 
 def display_search_results_export_tab(search_query: str, search_type: str, page: int = 1, items_per_page: int = 10):
@@ -5052,11 +5133,12 @@ def create_document_editing_tab():
                     api_name_input = gr.Dropdown(
                         choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
                                  "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                        value=None,
+                        value="Mistral",
                         label="API for Grammar Check"
                     )
                     api_key_input = gr.Textbox(label="API Key (if not set in config.txt)", placeholder="Enter your API key here",
-                                                   type="password")
+                                                   type="password",
+                                               value=f"{MISTRAL_TOKEN}")
                     check_grammar_button = gr.Button("Check Grammar and Style")
 
                 with gr.Column():
@@ -5081,11 +5163,12 @@ def create_document_editing_tab():
                     api_name_input = gr.Dropdown(
                         choices=[None, "Local-LLM", "OpenAI", "Anthropic", "Cohere", "Groq", "DeepSeek", "Mistral", "OpenRouter",
                                  "Llama.cpp", "Kobold", "Ooba", "Tabbyapi", "VLLM", "HuggingFace"],
-                        value=None,
+                        value="Mistral",
                         label="API for Grammar Check"
                     )
                     api_key_input = gr.Textbox(label="API Key (if not set in config.txt)", placeholder="Enter your API key here",
-                                                   type="password")
+                                                   type="password",
+                                               value=f"{MISTRAL_TOKEN}")
                     adjust_btn = gr.Button("Adjust Tone")
 
                 with gr.Column():