Update app.py

app.py

@@ -46,18 +46,17 @@ def process_pdf(file):
         )
         splits = text_splitter.split_documents(docs)
         
-        # Create FAISS index
         texts = [doc.page_content for doc in splits]
-        embeddings = embedding_model.embed_documents(texts)
+        metadatas = [doc.metadata for doc in splits]
         
-        # Convert to numpy array
-        embeddings_np = np.array(embeddings).astype('float32')
+        # Generate embeddings
+        embeddings = embedding_model.embed_documents(texts)
         
-        # Create FAISS index directly
-        vectorstore = FAISS.from_embeddings(
-            text_embeddings=list(zip(texts, embeddings)),
-            embedding=embedding_model,
-            metadatas=[doc.metadata for doc in splits]
+        # Create FAISS index
+        vectorstore = FAISS.from_texts(
+            texts=texts,
+            embedding=embedding_model.embed_query,
+            metadatas=metadatas
         )
         
         os.unlink(tmp_path)
@@ -73,16 +72,14 @@ def answer_question(question):
     try:
         docs = vectorstore.similarity_search(question, k=3)
         context = "\n\n".join([doc.page_content for doc in docs])
-        
-        # Simple QA implementation
-        answer = f"Relevant content from document:\n{context[:2000]}..."
-        sources = "\n📄 Sources:\n" + "\n".join([f"- Page {doc.metadata['page']+1}" for doc in docs])
-        return f"{answer}\n{sources}"
+        sources = "\n📄 Sources:\n" + "\n".join([f"- Page {doc.metadata.get('page', 'N/A') + 1}" for doc in docs])
+        answer = f"Relevant content from document:\n{context[:2000]}...\n{sources}"
+        return answer
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
 with gr.Blocks() as app:
-    gr.Markdown("# PDF Question Answering System")
+    gr.Markdown("# 📄 PDF Question Answering System")
     
     with gr.Row():
         file_input = gr.File(label="Upload PDF", type="binary")
@@ -90,10 +87,10 @@ with gr.Blocks() as app:
         status = gr.Textbox(label="Status")
     
     question = gr.Textbox(label="Your Question")
-    answer = gr.Textbox(label="Answer", interactive=False, lines=5)
+    answer = gr.Textbox(label="Answer", interactive=False, lines=10)
     ask_btn = gr.Button("Get Answer")
 
     upload_btn.click(process_pdf, inputs=file_input, outputs=status)
     ask_btn.click(answer_question, inputs=question, outputs=answer)
 
-app.launch()
+app.launch()