Upload app.py

app.py

@@ -148,6 +148,9 @@ def create_new_database(file_content: str, db_name: str, password: str, progress
 def generate_response(user_input: str, db_name: str) -> Optional[str]:
     """Generate response using Qwen2.5 MAX"""
     try:
+        if not db_name:
+            return "Please select a database to chat with."
+
         db_path = os.path.join(DATABASE_DIR, db_name)
         if not os.path.exists(db_path):
             return f"Database '{db_name}' does not exist."
@@ -212,6 +215,15 @@ embeddings = LocalEmbeddings(model)
 with gr.Blocks() as app:
     gr.Markdown("# Local Tech Knowledge Assistant")
 
+    # Shared state for database list
+    db_list_state = gr.State([])
+
+    def update_db_list():
+        """Update the list of available databases"""
+        if not os.path.exists(DATABASE_DIR):
+            return []
+        return [name for name in os.listdir(DATABASE_DIR) if os.path.isdir(os.path.join(DATABASE_DIR, name))]
+
     with gr.Tab("Create Database"):
         gr.Markdown("## Create a New FAISS Database")
         file_input = gr.File(label="Upload .txt File")
@@ -234,12 +246,16 @@ with gr.Blocks() as app:
             else:
                 return "Invalid file format. Please upload a .txt file."
             
-            return create_new_database(file_content, db_name, password, progress)
+            result = create_new_database(file_content, db_name, password, progress)
+            if "created successfully" in result:
+                # Update the database list
+                return result, update_db_list()
+            return result, None
         
         create_button.click(
             handle_create,
             inputs=[file_input, db_name_input, password_input],
-            outputs=create_output
+            outputs=[create_output, db_list_state]
         )
 
     with gr.Tab("Chat with Database"):
@@ -249,12 +265,9 @@ with gr.Blocks() as app:
         msg = gr.Textbox(label="Votre question")
         clear = gr.ClearButton([msg, chatbot])
         
-        def update_db_list():
-            if not os.path.exists(DATABASE_DIR):
-                return []
-            return [name for name in os.listdir(DATABASE_DIR) if os.path.isdir(os.path.join(DATABASE_DIR, name))]
-        
         def chat_response(message: str, db_name: str, history: List[Tuple[str, str]]):
+            if not db_name:
+                return "", history + [("System", "Please select a database to chat with.")]
             response = generate_response(message, db_name)
             return "", history + [(message, response or "Erreur de génération - Veuillez réessayer.")]
         
@@ -265,8 +278,15 @@ with gr.Blocks() as app:
             queue=True
         )
         
-        # Update database list on page load
+        # Update dropdown on page load
         db_select.choices = update_db_list()
 
+        # Update dropdown when db_list_state changes
+        db_list_state.change(
+            lambda dbs: gr.Dropdown.update(choices=dbs),
+            inputs=db_list_state,
+            outputs=db_select
+        )
+
 if __name__ == "__main__":
     app.launch(server_name="0.0.0.0", server_port=7860)