Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import fitz  # PyMuPDF
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-from langchain.vectorstores import Chroma
+from langchain_community.vectorstores import Chroma
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 import os
@@ -33,9 +33,9 @@ def get_llm_response(input_prompt, content, prompt):
 
 
 # Function to extract text from PDF file
-def extract_text_from_pdf(file):
+def extract_text_from_pdf(file_path):
     try:
-        doc = fitz.open(stream=file.read(), filetype="pdf")
+        doc = fitz.open(file_path)
         text = ""
         for page in doc:
             text += page.get_text()
@@ -47,7 +47,7 @@ def extract_text_from_pdf(file):
 def process_pdf(uploaded_file, prompt):
     if uploaded_file is not None:
         # Extract text from uploaded PDF file
-        pdf_text = extract_text_from_pdf(uploaded_file)
+        pdf_text = extract_text_from_pdf(uploaded_file.name)
         if pdf_text:
             try:
                 # Create embeddings
@@ -92,8 +92,8 @@ def process_pdf(uploaded_file, prompt):
 def main():
     gr.Interface(
         fn=process_pdf,
-        inputs=[gr.components.File(type="file", label="Upload PDF File"),
-                gr.components.Textbox(lines=2, placeholder="Ask a Question")],
+        inputs=[gr.File(type="filepath", label="Upload PDF File"),
+                gr.Textbox(lines=2, placeholder="Ask a Question")],
         outputs="text",
         title="PDF Chatbot",
         description="Upload a PDF file and ask questions about its content."