Update app.py

app.py

@@ -61,9 +61,7 @@ def load_model():
 template = """
 You are an intelligent assistant specialized in document analysis.
 Use the following information from PDF files to answer the question:
-
 {context}
-
 Question: {question}
 Answer (in detail and in clear language):
 """
@@ -91,25 +89,28 @@ def respond(message, chat_history):
     except Exception as e:
         return chat_history + [(message, f"An error occurred: {str(e)}")]
 
-# Handle file uploads
+    # Event handling
 def handle_upload(files):
-    global qa_chain
-    
-    try:
-        file_contents = [file.read() for file in files]
-        retriever = process_pdfs(file_contents)
-        llm = load_model()
-        
-        qa_chain = RetrievalQA.from_chain_type(
-            llm=llm,
-            retriever=retriever,
-            chain_type="stuff",
-            chain_type_kwargs={"prompt": prompt},
-            return_source_documents=False
-        )
-        return "Files uploaded successfully! You can now ask questions."
-    except Exception as e:
-        return f"Error uploading files: {str(e)}"
+        global qa_chain
+        try:
+            retriever = process_pdfs(files)
+            llm = load_model()
+            
+            qa_chain = RetrievalQA.from_chain_type(
+                llm=llm,
+                retriever=retriever,
+                chain_type="stuff",
+                chain_type_kwargs={
+                    "prompt": PromptTemplate(
+                        template=template,
+                        input_variables=["context", "question"]
+                    )
+                },
+                return_source_documents=False
+            )
+            return "Files uploaded successfully!"
+        except Exception as e:
+            return f"Error uploading files: {str(e)}"
 
 # UI
 with gr.Blocks(title="Smart Document Assistant", theme=gr.themes.Default()) as demo:
@@ -139,28 +140,7 @@ with gr.Blocks(title="Smart Document Assistant", theme=gr.themes.Default()) as d
     
     clear_btn = gr.Button("Clear Chat")
     
-    # Event handling
-    def handle_upload(files):
-        global qa_chain
-        try:
-            retriever = process_pdfs(files)
-            llm = load_model()
-            
-            qa_chain = RetrievalQA.from_chain_type(
-                llm=llm,
-                retriever=retriever,
-                chain_type="stuff",
-                chain_type_kwargs={
-                    "prompt": PromptTemplate(
-                        template=template,
-                        input_variables=["context", "question"]
-                    )
-                },
-                return_source_documents=False
-            )
-            return "Files uploaded successfully!"
-        except Exception as e:
-            return f"Error uploading files: {str(e)}"
+
     
     file_upload.change(
         handle_upload,