Update app.py

app.py

@@ -53,7 +53,6 @@ file_extractor = {
 # Markdown content definitions
 description = """
 ## Welcome to DocBot 📄🤖
-
 DocBot is an intelligent document analysis tool that can help you extract insights from various document formats including:
 - PDF documents
 - Word documents (.docx, .doc)
@@ -63,19 +62,16 @@ DocBot is an intelligent document analysis tool that can help you extract insigh
 - PowerPoint presentations
 - HTML files
 - Images with text (JPG, PNG, WebP, SVG)
-
 Simply upload your document, select your preferred embedding model and LLM, then start asking questions!
 """
 
 guide = """
 ### How to Use DocBot:
-
 1. **Upload Document**: Choose any supported file format
 2. **Select Embedding Model**: Choose from available embedding models (BAAI/bge-small-en-v1.5 is recommended for most cases)
 3. **Submit**: Click submit to process your document
 4. **Select LLM**: Choose your preferred language model
 5. **Ask Questions**: Start chatting with your document!
-
 ### Tips:
 - Smaller embedding models (like bge-small-en-v1.5) are faster but may be less accurate
 - Larger models provide better understanding but take more time
@@ -165,6 +161,10 @@ def encode_image_safe(image_path):
         pass
     return ""
 
+# Clear function for file processing components
+def clear_file_components():
+    return None, embed_models[0], ""
+
 # Encode the images (with fallback for missing images)
 github_logo_encoded = encode_image_safe("Images/github-logo.png")
 linkedin_logo_encoded = encode_image_safe("Images/linkedin-logo.png")
@@ -208,7 +208,7 @@ with gr.Blocks(
                         
                         with gr.Row():
                             btn = gr.Button("🚀 Process Document", variant='primary', size="lg")
-                            clear = gr.ClearButton("🗑️ Clear", size="lg")
+                            clear_btn = gr.Button("🗑️ Clear", size="lg")
                         
                         output = gr.Textbox(
                             label='Processing Status', 
@@ -246,7 +246,7 @@ with gr.Blocks(
                     
                     with gr.Row():
                         submit_btn = gr.Button("Send", variant="primary")
-                        clear_chat = gr.ClearButton([msg, chatbot], value="Clear Chat")
+                        clear_chat_btn = gr.Button("Clear Chat")
 
     # Add footer if images exist
     if any([github_logo_encoded, linkedin_logo_encoded, website_logo_encoded]):
@@ -261,6 +261,9 @@ with gr.Blocks(
         history.append([message, response])
         return history, ""
     
+    def clear_chat():
+        return [], ""
+    
     # Event bindings
     llm_model_dropdown.change(
         fn=set_llm_model, 
@@ -286,10 +289,15 @@ with gr.Blocks(
         outputs=[chatbot, msg]
     )
     
-    clear.click(
-        lambda: [None, None, ""], 
+    clear_btn.click(
+        fn=clear_file_components,
         outputs=[file_input, embed_model_dropdown, output]
     )
+    
+    clear_chat_btn.click(
+        fn=clear_chat,
+        outputs=[chatbot, msg]
+    )
 
 # Launch the demo
 if __name__ == "__main__":