Update app.py

app.py

@@ -47,22 +47,22 @@ import streamlit as st
 #     return chain
 
 # # Function to generate a response based on the chat history and query
-# def generate_response(history, query, btn):
-#     global COUNT, N, chat_history, chain 
-#     if not btn:
-#         raise gr.Error(message='Upload a PDF')
+def generate_response(history, query, btn):
+    global COUNT, N, chat_history, chain 
+    if not btn:
+        raise gr.Error(message='Upload a PDF')
 
-#     if COUNT == 0:
-#         chain = process_file(btn)
-#         COUNT += 1
+    if COUNT == 0:
+        chain = process_file(btn)
+        COUNT += 1
 
-#     result = chain({"question": query, 'chat_history': chat_history}, return_only_outputs=True)
-#     chat_history += [(query, result["answer"])]
-#     N = list(result['source_documents'][0])[1][1]['page']
+    result = chain({"question": query, 'chat_history': chat_history}, return_only_outputs=True)
+    chat_history += [(query, result["answer"])]
+    N = list(result['source_documents'][0])[1][1]['page']
 
-#     for char in result['answer']:
-#         history[-1][-1] += char  # Update the last response
-#         yield history, ''  
+    for char in result['answer']:
+        history[-1][-1] += char  # Update the last response
+        yield history, ''  
 
 # # Function to render a specific page of a PDF file as an image
 # def render_file(file):
@@ -147,6 +147,7 @@ import streamlit as st
 
 # demo.launch(server_port=7861) 
 
+
 def add_text(history, text):
     if not text:
         raise gr.Error('Enter text')