Cleanup: Remove list documents feature and save visualization, keep download feature

app.py

@@ -433,20 +433,21 @@ def process_file_upload(files):
     
     return "\n".join(file_info)
 
-def list_documents():
-    """List all indexed documents"""
-    try:
-        docs = document_assistant.get_all_documents()
-        if not docs:
-            return "No documents indexed yet."
-        
-        result = "Indexed Documents:\n\n"
-        for doc in docs:
-            result += f"- {doc['filename']} ({doc['file_type']})\n"
-        
-        return result
-    except Exception as e:
-        return f"Error listing documents: {str(e)}"
+# Function commented out as it's no longer used
+# def list_documents():
+#     """List all indexed documents"""
+#     try:
+#         docs = document_assistant.get_all_documents()
+#         if not docs:
+#             return "No documents indexed yet."
+#         
+#         result = "Indexed Documents:\n\n"
+#         for doc in docs:
+#             result += f"- {doc['filename']} ({doc['file_type']})\n"
+#         
+#         return result
+#     except Exception as e:
+#         return f"Error listing documents: {str(e)}"
 
 def clear_context():
     """Clear the current context"""
@@ -775,7 +776,6 @@ with gr.Blocks(title="LLM Powered Database Chatbot") as demo:
                 
                 with gr.Row():
                     clear_viz_btn = gr.Button("🗑️ Clear Visualization")
-                    save_viz_btn = gr.Button("💾 Save Visualization")
                     download_btn = gr.Button("📥 Download Visualization")
                 
                 save_status = gr.Textbox(label="Save Status", visible=False)
@@ -793,29 +793,6 @@ with gr.Blocks(title="LLM Powered Database Chatbot") as demo:
         def clear_visualization():
             return "", ""
         
-        def save_visualization(viz_html):
-            if not viz_html:
-                return "No visualization to save", gr.update(visible=True)
-            
-            try:
-                # Extract the base64 image data from the HTML
-                img_data_match = re.search(r'src=\'data:image/png;base64,([^\']+)\'', viz_html)
-                
-                if img_data_match:
-                    # Get the base64 data
-                    img_data = img_data_match.group(1)
-                    
-                    # Create a downloadable file
-                    timestamp = time.strftime("%Y%m%d_%H%M%S")
-                    filename = f"visualization_{timestamp}.png"
-                    
-                    # Return success message with download link
-                    return f"Visualization ready for download", gr.update(visible=True)
-                else:
-                    return "Could not extract image data", gr.update(visible=True)
-            except Exception as e:
-                return f"Error preparing visualization for download: {str(e)}", gr.update(visible=True)
-
         def download_visualization(viz_html):
             if not viz_html:
                 return None
@@ -848,12 +825,6 @@ with gr.Blocks(title="LLM Powered Database Chatbot") as demo:
             outputs=[visualization_output, current_visualization]
         )
         
-        save_viz_btn.click(
-            save_visualization,
-            inputs=[current_visualization],
-            outputs=[save_status, save_status]
-        )
-        
         download_btn.click(
             download_visualization,
             inputs=[current_visualization],
@@ -917,61 +888,6 @@ with gr.Blocks(title="LLM Powered Database Chatbot") as demo:
             inputs=[],
             outputs=[upload_output]
         )
-        
-        list_docs_button = gr.Button("List Indexed Documents")
-        docs_output = gr.Textbox(label="Indexed Documents")
-        
-        list_docs_button.click(
-            list_documents,
-            inputs=[],
-            outputs=[docs_output]
-        )
-    
-    # Settings tab commented out as it's not currently used
-    # with gr.Tab("Settings"):
-    #     with gr.Row():
-    #         gr.Markdown("## Database Management")
-    #         flush_db_btn = gr.Button("🗑️ Flush All Databases", variant="stop", scale=1)
-    #     
-    #     flush_result = gr.Textbox(label="Flush Result")
-    #     
-    #     flush_db_btn.click(
-    #         flush_databases,
-    #         inputs=[],
-    #         outputs=[flush_result]
-    #     )
-    #     
-    #     gr.Markdown("## System Settings")
-    #     api_key = gr.Textbox(
-    #         label="Groq API Key",
-    #         placeholder="Enter your Groq API key",
-    #         type="password",
-    #         value=os.getenv("GROQ_API_KEY", "")
-    #     )
-    #     save_btn = gr.Button("Save Settings")
-    #     
-    #     def save_settings(key):
-    #         try:
-    #             os.environ["GROQ_API_KEY"] = key
-    #             return "Settings saved!"
-    #         except Exception as e:
-    #             return f"Error saving settings: {str(e)}"
-    #     
-    #     save_btn.click(
-    #         save_settings,
-    #         inputs=[api_key],
-    #         outputs=[gr.Textbox(label="Status")]
-    #     )
-    #     
-    #     gr.Markdown("## Debugging")
-    #     test_viz_btn = gr.Button("Test Visualization")
-    #     test_viz_output = gr.HTML(label="Test Visualization")
-    #     
-    #     test_viz_btn.click(
-    #         create_test_html_visualization,
-    #         inputs=[],
-    #         outputs=[test_viz_output]
-    #     )
 
 # Launch the app
 if __name__ == "__main__":