Update app.py

app.py

@@ -190,23 +190,26 @@ def split_text(text, chunk_size=500):
     print("splitting")
     return [text[i:i+chunk_size] for i in range(0, len(text), chunk_size)]
 
-def chatbot(user_question):
+def chatbot(doc_presence, user_question):
     """Processes the PDF and answers the user's question."""
     print("chatbot start")
-   
-    # retrieve the document relevant to the query
-    doc = retrieve_document(user_question)           
-    
-    if doc:
-        print(f"found doc:\n{doc}\n")
-        # Split into smaller chunks
-        chunks = split_text(doc)
-      
-        # Use only the first chunk (to optimize token usage)
-        prompt = f"Based on this document, answer the question:\n\nDocument:\n{chunks[0]}\n\nQuestion: {user_question}"
-        print(f"prompt:\n{prompt}")
+
+    if doc_presence:
+        # retrieve the document relevant to the query
+        doc = retrieve_document(user_question)           
+        
+        if doc:
+            print(f"found doc:\n{doc}\n")
+            # Split into smaller chunks
+            chunks = split_text(doc)
+          
+            # Use only the first chunk (to optimize token usage)
+            prompt = f"Based on this document, answer the question:\n\nDocument:\n{chunks[0]}\n\nQuestion: {user_question}"
+            print(f"prompt:\n{prompt}")
+        else:
+                  prompt=user_question
     else:
-              prompt=user_question
+        prompt=user_question
 
     try:
             print("asking", prompt)
@@ -232,7 +235,7 @@ iface = gr.TabbedInterface(
     [
         gr.Interface(
             fn=chatbot,
-            inputs=gr.Textbox(label="Ask a Question"),
+            inputs=[gr.Textbox(label="Ask a Question"), "text"]
             outputs=gr.Textbox(label="Answer"),
             title="PDF Q&A Chatbot (Powered by Together.AI)",
         ),