Update app.py

app.py

@@ -312,6 +312,261 @@ def create_download_link_with_cache(
             st.error(f"Error creating download link: {str(e)}")
             return ""
 
+
+
+# ---
+def display_voice_tab():
+    """Display voice input tab with TTS settings."""
+    st.subheader("🎤 Voice Input")
+    
+    # Voice Settings Section
+    st.markdown("### 🎤 Voice Settings")
+    selected_voice = st.selectbox(
+        "Select TTS Voice:",
+        options=EDGE_TTS_VOICES,
+        index=EDGE_TTS_VOICES.index(st.session_state['tts_voice'])
+    )
+    
+    # Audio Format Selection
+    st.markdown("### 🔊 Audio Format")
+    selected_format = st.radio(
+        "Choose Audio Format:",
+        options=["MP3", "WAV"],
+        index=0
+    )
+
+    # Update session state if settings change
+    if selected_voice != st.session_state['tts_voice']:
+        st.session_state['tts_voice'] = selected_voice
+        st.rerun()
+    if selected_format.lower() != st.session_state['audio_format']:
+        st.session_state['audio_format'] = selected_format.lower()
+        st.rerun()
+
+    # Text Input Area
+    user_text = st.text_area("💬 Message:", height=100)
+    user_text = user_text.strip().replace('\n', ' ')
+
+    # Send Button
+    if st.button("📨 Send"):
+        process_voice_input(user_text)
+
+    # Chat History
+    st.subheader("📜 Chat History")
+    for c in st.session_state.chat_history:
+        st.write("**You:**", c["user"])
+        st.write("**Response:**", c["claude"])
+
+def display_arxiv_tab():
+    """Display ArXiv search tab with options."""
+    st.subheader("🔍 Query ArXiv")
+    q = st.text_input("🔍 Query:", key="arxiv_query")
+    
+    # Options Section
+    st.markdown("### 🎛 Options")
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        vocal_summary = st.checkbox("🎙 Short Audio", value=True, 
+                                  key="option_vocal_summary")
+        extended_refs = st.checkbox("📜 Long Refs", value=False, 
+                                  key="option_extended_refs")
+        
+    with col2:
+        titles_summary = st.checkbox("🔖 Titles Only", value=True, 
+                                   key="option_titles_summary")
+        full_audio = st.checkbox("📚 Full Audio", value=False, 
+                               key="option_full_audio")
+    
+    full_transcript = st.checkbox("🧾 Full Transcript", value=False, 
+                                key="option_full_transcript")
+    
+    if q and st.button("🔍 Run Search"):
+        st.session_state.last_query = q
+        result, timings = perform_ai_lookup(
+            q,
+            vocal_summary=vocal_summary,
+            extended_refs=extended_refs,
+            titles_summary=titles_summary,
+            full_audio=full_audio
+        )
+        
+        if full_transcript:
+            create_file(q, result, "md")
+
+def display_media_tab():
+    """Display media gallery tab with audio, images, and video."""
+    st.header("📸 Media Gallery")
+
+    # Create tabs for different media types
+    tabs = st.tabs(["🎵 Audio", "🖼 Images", "🎥 Video"])
+    
+    # Audio Files Tab
+    with tabs[0]:
+        st.subheader("🎵 Audio Files")
+        audio_files = glob.glob("*.mp3") + glob.glob("*.wav")
+        
+        if audio_files:
+            for audio_file in audio_files:
+                with st.expander(os.path.basename(audio_file)):
+                    st.audio(audio_file)
+                    ext = os.path.splitext(audio_file)[1].replace('.', '')
+                    dl_link = get_download_link(audio_file, file_type=ext)
+                    st.markdown(dl_link, unsafe_allow_html=True)
+        else:
+            st.write("No audio files found.")
+    
+    # Images Tab
+    with tabs[1]:
+        st.subheader("🖼 Image Files")
+        image_files = glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.jpeg")
+        
+        if image_files:
+            cols = st.slider("Columns:", 1, 5, 3, key="cols_images")
+            image_cols = st.columns(cols)
+            
+            for i, img_file in enumerate(image_files):
+                with image_cols[i % cols]:
+                    try:
+                        img = Image.open(img_file)
+                        st.image(img, use_column_width=True)
+                    except Exception as e:
+                        st.error(f"Error loading image {img_file}: {str(e)}")
+        else:
+            st.write("No images found.")
+    
+    # Video Tab
+    with tabs[2]:
+        st.subheader("🎥 Video Files")
+        video_files = glob.glob("*.mp4") + glob.glob("*.mov") + glob.glob("*.avi")
+        
+        if video_files:
+            for video_file in video_files:
+                with st.expander(os.path.basename(video_file)):
+                    st.video(video_file)
+        else:
+            st.write("No videos found.")
+
+def display_editor_tab():
+    """Display text editor tab with file management."""
+    st.subheader("📝 Text Editor")
+
+    # File Management Section
+    st.markdown("### 📂 File Management")
+    
+    # File Selection
+    md_files = glob.glob("*.md")
+    selected_file = st.selectbox(
+        "Select file to edit:",
+        ["New File"] + md_files,
+        key="file_selector"
+    )
+
+    # Edit Area
+    if selected_file == "New File":
+        new_filename = st.text_input("New filename (without extension):")
+        file_content = st.text_area("Content:", height=300)
+        
+        if st.button("💾 Save File"):
+            if new_filename:
+                try:
+                    with open(f"{new_filename}.md", 'w', encoding='utf-8') as f:
+                        f.write(file_content)
+                    st.success(f"File {new_filename}.md saved successfully!")
+                    st.session_state.should_rerun = True
+                except Exception as e:
+                    st.error(f"Error saving file: {str(e)}")
+            else:
+                st.warning("Please enter a filename.")
+    else:
+        try:
+            # Load existing file content
+            with open(selected_file, 'r', encoding='utf-8') as f:
+                file_content = f.read()
+            
+            # Edit existing file
+            edited_content = st.text_area(
+                "Edit content:",
+                value=file_content,
+                height=300
+            )
+            
+            col1, col2 = st.columns(2)
+            with col1:
+                if st.button("💾 Save Changes"):
+                    try:
+                        with open(selected_file, 'w', encoding='utf-8') as f:
+                            f.write(edited_content)
+                        st.success("Changes saved successfully!")
+                    except Exception as e:
+                        st.error(f"Error saving changes: {str(e)}")
+            
+            with col2:
+                if st.button("🗑 Delete File"):
+                    try:
+                        os.remove(selected_file)
+                        st.success(f"File {selected_file} deleted successfully!")
+                        st.session_state.should_rerun = True
+                    except Exception as e:
+                        st.error(f"Error deleting file: {str(e)}")
+                        
+        except Exception as e:
+            st.error(f"Error loading file {selected_file}: {str(e)}")
+
+def display_settings_tab():
+    """Display application settings tab."""
+    st.subheader("⚙️ Settings")
+    
+    # General Settings
+    st.markdown("### 🔧 General Settings")
+    
+    # Theme Selection
+    theme = st.selectbox(
+        "Color Theme:",
+        ["Dark", "Light", "Custom"],
+        index=0
+    )
+    
+    if theme == "Custom":
+        st.color_picker("Primary Color:", "#1E1E1E")
+        st.color_picker("Secondary Color:", "#2D2D2D")
+    
+    # Performance Settings
+    st.markdown("### ⚡ Performance Settings")
+    
+    # Cache Settings
+    cache_size = st.slider(
+        "Maximum Cache Size (MB):",
+        0, 1000, 100
+    )
+    
+    if st.button("Clear Cache"):
+        st.session_state['audio_cache'] = {}
+        st.session_state['paper_cache'] = {}
+        st.session_state['download_link_cache'] = {}
+        st.success("Cache cleared successfully!")
+    
+    # API Settings
+    st.markdown("### 🔑 API Settings")
+    
+    # Show/hide API keys
+    show_keys = st.checkbox("Show API Keys")
+    if show_keys:
+        st.text_input("OpenAI API Key:", value=openai_api_key)
+        st.text_input("Anthropic API Key:", value=anthropic_key)
+    
+    # Save Settings
+    if st.button("💾 Save Settings"):
+        st.success("Settings saved successfully!")
+        st.session_state.should_rerun = True
+
+
+
+
+
+
+
+
 # ─────────────────────────────────────────────────────────
 # 4. PAPER PROCESSING & DISPLAY
 # ─────────────────────────────────────────────────────────