Update app.py

app.py

@@ -3,10 +3,10 @@ import openai
 import fitz  # PyMuPDF for PDF processing
 import base64
 
-# Variable to store the API key
+# Variable to store API key
 api_key = ""
 
-# Function to update the API key
+# Function to update API key
 def set_api_key(key):
     global api_key
     api_key = key
@@ -87,9 +87,9 @@ def pdf_chat(pdf_file, text_query, temperature, top_p, max_output_tokens):
     ]
     return query_openai(messages, temperature, top_p, max_output_tokens)
 
-# Function to clear the chat (Resets input and output)
+# Function to clear the chat (Fix: Returns the correct number of outputs)
 def clear_chat():
-    return "", "", "", "", "", "", "", 1.0, 1.0, 1024
+    return "", "", "", "", "", "", "", None, "", None, "", 1.0, 1.0, 1024
 
 # Gradio UI Layout
 with gr.Blocks() as demo:
@@ -140,14 +140,14 @@ with gr.Blocks() as demo:
     image_button.click(image_chat, [image_upload, image_text_query, temperature, top_p, max_output_tokens], image_output)
     pdf_button.click(pdf_chat, [pdf_upload, pdf_text_query, temperature, top_p, max_output_tokens], pdf_output)
 
-    # Fix: Clear button resets only necessary fields
+    # Fix: Clear button resets all necessary fields correctly
     clear_button.click(
         clear_chat,
         outputs=[
             image_url, image_query, image_url_output, 
             text_query, text_output, 
             image_text_query, image_output, 
-            pdf_text_query, pdf_output, 
+            pdf_upload, pdf_text_query, pdf_output, 
             temperature, top_p, max_output_tokens
         ]
     )