Upload app.py

app.py

@@ -18,7 +18,7 @@ QWEN_API_URL = "Qwen/Qwen2.5-Max-Demo"  # Gradio API for Qwen2.5 chat
 CHUNK_SIZE = 800
 TOP_K_RESULTS = 150
 SIMILARITY_THRESHOLD = 0.4
-PASSWORD = "abc12345"
+PASSWORD_HASH = "abc12345"  # Replace with hashed password in production
 
 BASE_SYSTEM_PROMPT = """
 Répondez en français selon ces règles :
@@ -76,7 +76,7 @@ def split_text_into_chunks(text: str) -> List[str]:
             chunk.rfind('\n\n')
         )
         
-        if last_punct != -1 and (end - start) > CHUNK_SIZE//2:
+        if last_punct != -1 and (end - start) > CHUNK_SIZE // 2:
             end = start + last_punct + 1
         
         chunks.append(text[start:end].strip())
@@ -105,7 +105,7 @@ def initialize_vector_store(embeddings: Embeddings, db_name: str) -> FAISS:
 
 def create_new_database(file_content: str, db_name: str, password: str) -> str:
     """Create a new FAISS database from uploaded file"""
-    if password != PASSWORD:
+    if password != PASSWORD_HASH:
         return "Incorrect password. Database creation failed."
 
     if not file_content.strip():
@@ -149,7 +149,7 @@ def generate_response(user_input: str, db_name: str) -> Optional[str]:
         # Contextual search
         docs_scores = vector_store.similarity_search_with_score(
             user_input, 
-            k=TOP_K_RESULTS*3
+            k=TOP_K_RESULTS * 3
         )
         
         # Filter results
@@ -191,7 +191,7 @@ def generate_response(user_input: str, db_name: str) -> Optional[str]:
         logging.error(f"Generation error: {str(e)}", exc_info=True)
         return None
 
-# Initialize models
+# Initialize models and vector store
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = SentenceTransformer("cnmoro/snowflake-arctic-embed-m-v2.0-cpu", device=device, trust_remote_code=True)
 embeddings = LocalEmbeddings(model)
@@ -199,7 +199,7 @@ embeddings = LocalEmbeddings(model)
 # Gradio interface
 with gr.Blocks() as app:
     gr.Markdown("# Local Tech Knowledge Assistant")
-    
+
     with gr.Tab("Create Database"):
         gr.Markdown("## Create a New FAISS Database")
         file_input = gr.File(label="Upload .txt File")
@@ -213,7 +213,11 @@ with gr.Blocks() as app:
                 return "Please provide all required inputs."
             
             # Read file content
-            file_content = file.decode("utf-8")
+            try:
+                file_content = file.read().decode("utf-8")
+            except Exception as e:
+                return f"Error reading file: {str(e)}"
+            
             return create_new_database(file_content, db_name, password)
         
         create_button.click(
@@ -221,7 +225,7 @@ with gr.Blocks() as app:
             inputs=[file_input, db_name_input, password_input],
             outputs=create_output
         )
-    
+
     with gr.Tab("Chat with Database"):
         gr.Markdown("## Chat with Existing Databases")
         db_select = gr.Dropdown(choices=[], label="Select Database")