Update other_function.py

other_function.py

@@ -10,7 +10,7 @@ import os
 from pypdf import PdfReader
 from ai71 import AI71
 import os
-
+import PyPDF2
 import pandas as pd
 
 from inference_sdk import InferenceHTTPClient
@@ -19,6 +19,7 @@ UPLOAD_FOLDER = '/code/uploads'
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
 
+pdf_text=''
 AI71_API_KEY = os.environ.get('AI71_API_KEY')
 def generate_response(query,chat_history):
     response = ''
@@ -169,7 +170,7 @@ def download_and_save_as_txt(url, account_sid, auth_token):
             file.write(response.content)
         
         print(f"Media downloaded successfully and saved as {txt_filepath}")
-        return txt_filepath
+        return txt_filepath,extract_text_from_pdf(pdf_path)
 
     except requests.exceptions.HTTPError as err:
         print(f"HTTP error occurred: {err}")
@@ -193,4 +194,31 @@ def download_file(url, extension):
         print(f"HTTP error occurred: {err}")
     except Exception as err:
         print(f"An error occurred: {err}")
-        return None
+        return None
+
+
+def extract_text_from_pdf(pdf_path):
+    global pdf_text
+    with open(pdf_path, 'rb') as file:
+        reader = PyPDF2.PdfReader(file)
+        pdf_text = ''
+        for page_num in range(len(reader.pages)):
+            page = reader.pages[page_num]
+            pdf_text += page.extract_text()
+        return pdf_text
+
+
+def respond_pdf(extracted_text,query):
+    if 
+    res = ''
+    for chunk in AI71(AI71_API_KEY).chat.completions.create(
+        model="tiiuae/falcon-11b",
+        messages=[
+            {"role": "system", "content": "You are a pdf answering assistant."},
+            {"role": "user", "content": f"Content:{extracted_text},Query:{query}"},
+        ],
+        stream=True,
+    ):
+        if chunk.choices[0].delta.content:
+            res += chunk.choices[0].delta.content
+    return ( res.replace("User:",'').strip())