Update app.py

app.py

@@ -32,6 +32,9 @@ def query_openai(messages, temperature, top_p, max_output_tokens):
 
 # Function to process image URL input
 def image_url_chat(image_url, text_query, temperature, top_p, max_output_tokens):
+    if not image_url or not text_query:
+        return "Please provide an image URL and a query."
+
     messages = [
         {"role": "user", "content": [
             {"type": "image_url", "image_url": {"url": image_url}},  # Corrected format
@@ -42,13 +45,16 @@ def image_url_chat(image_url, text_query, temperature, top_p, max_output_tokens)
 
 # Function to process text input
 def text_chat(text_query, temperature, top_p, max_output_tokens):
+    if not text_query:
+        return "Please enter a query."
+
     messages = [{"role": "user", "content": [{"type": "text", "text": text_query}]}]
     return query_openai(messages, temperature, top_p, max_output_tokens)
 
 # Function to process uploaded image input
 def image_chat(image_file, text_query, temperature, top_p, max_output_tokens):
-    if image_file is None:
-        return "Please upload an image."
+    if image_file is None or not text_query:
+        return "Please upload an image and provide a query."
 
     # Encode image as base64
     with open(image_file, "rb") as img:
@@ -66,8 +72,8 @@ def image_chat(image_file, text_query, temperature, top_p, max_output_tokens):
 
 # Function to process uploaded PDF input
 def pdf_chat(pdf_file, text_query, temperature, top_p, max_output_tokens):
-    if pdf_file is None:
-        return "Please upload a PDF."
+    if pdf_file is None or not text_query:
+        return "Please upload a PDF and provide a query."
 
     # Extract text from the first few pages
     doc = fitz.open(pdf_file)
@@ -81,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
+# Function to clear the chat (Resets input and output)
 def clear_chat():
-    return "", "", None, "", None, "", 1.0, 1.0, 1024
+    return "", "", "", "", "", "", "", 1.0, 1.0, 1024
 
 # Gradio UI Layout
 with gr.Blocks() as demo:
@@ -98,7 +104,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         temperature = gr.Slider(0, 2, value=1.0, step=0.1, label="Temperature")
         top_p = gr.Slider(0, 1, value=1.0, step=0.1, label="Top-P")
-        max_output_tokens = gr.Slider(0, 16384, value=2048, step=512, label="Max Output Tokens")
+        max_output_tokens = gr.Slider(0, 16384, value=1024, step=512, label="Max Output Tokens")
     
     with gr.Tabs():
         with gr.Tab("Image URL Chat"):
@@ -133,7 +139,18 @@ with gr.Blocks() as demo:
     text_button.click(text_chat, [text_query, temperature, top_p, max_output_tokens], text_output)
     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)
-    clear_button.click(clear_chat, outputs=[image_url, image_query, image_url_output, text_query, text_output, image_upload, image_text_query, image_output, pdf_upload, pdf_text_query, pdf_output, temperature, top_p, max_output_tokens])
+
+    # Fix: Clear button resets only necessary fields
+    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, 
+            temperature, top_p, max_output_tokens
+        ]
+    )
 
 # Launch Gradio App
 if __name__ == "__main__":