Update app.py

app.py

@@ -62,4 +62,66 @@ def extract_text_from_docx(docx_path):
         doc = Document(docx_path)
         text = "\n".join([para.text for para in doc.paragraphs])
     except Exception as e:
-        print
+        print(f"DOCX error: {e}")
+    return text
+
+# =============================================
+# DOCUMENT UPLOAD HANDLER
+# =============================================
+def upload_document(file):
+    ext = os.path.splitext(file.name)[-1].lower()
+    if ext == ".pdf":
+        text = extract_text_from_pdf(file.name)
+    elif ext == ".docx":
+        text = extract_text_from_docx(file.name)
+    else:
+        return "Unsupported file type"
+
+    embedding = get_embeddings(text)
+    index.add(embedding)
+    document_texts.append(text)
+
+    # Save updated index and texts
+    with open(index_path, "wb") as f:
+        pickle.dump(index, f)
+    with open(document_texts_path, "wb") as f:
+        pickle.dump(document_texts, f)
+
+    return "Document uploaded and indexed successfully!"
+
+# =============================================
+# SEMANTIC SEARCH HANDLER
+# =============================================
+def search_documents(query):
+    if not document_texts:
+        return "No documents indexed yet."
+
+    query_vector = get_embeddings(query)
+    scores, indices = index.search(query_vector, k=1)
+    best_match_idx = indices[0][0]
+
+    return f"**Best Match:**\n\n{document_texts[best_match_idx][:1000]}..."
+
+# =============================================
+# GRADIO INTERFACE
+# =============================================
+upload_interface = gr.Interface(
+    fn=upload_document,
+    inputs=gr.File(file_types=[".pdf", ".docx"]),
+    outputs="text",
+    title="Upload PDF/DOCX",
+    description="Upload a PDF or Word document to be indexed for semantic search."
+)
+
+search_interface = gr.Interface(
+    fn=search_documents,
+    inputs=gr.Textbox(placeholder="Enter your question or search query here..."),
+    outputs="markdown",
+    title="Semantic Search",
+    description="Search for content in uploaded documents using natural language."
+)
+
+app = gr.TabbedInterface([upload_interface, search_interface], ["Upload Document", "Search Document"])
+
+if __name__ == "__main__":
+    app.launch()