clear btn

app.py

@@ -8,7 +8,6 @@ import gradio as gr
 # import shutil
 import tempfile
 
-# default model
 repo_id = "mistralai/Mistral-7B-Instruct-v0.1"
 model_type = 'text-generation'
 
@@ -25,15 +24,15 @@ def add_text(history, text):
 
 
 def hasFile(history):
+    pdf_files = 0
     for user_prompt, bot_response in history:
         if '.pdf' in user_prompt.lower():
-            return True
+            pdf_files += 1
 
-    return False
+    return pdf_files
 
 def modelChanged(history, drop):
 
-
     history = history + [(f'===> {drop}', None)]
     return history, drop
 
@@ -76,6 +75,11 @@ def copy_pdf(source_path, destination_path):
 
 def add_file(history, file):
 
+    pdf_files = hasFile(history)
+    if pdf_files + 1 >= 4:
+        history = history + [("%s!!!"%os.path.basename(file), None)]
+        return history
+
     file_path = os.path.join(temp_dir.name, os.path.basename(file))
     # shutil.copyfile(file.name, file_path) # <---Asynchronous
     copy_pdf(file.name, file_path)
@@ -84,6 +88,9 @@ def add_file(history, file):
     history = history + [(os.path.basename(file), None)]
     return history
 
+def clearClick():
+    print("clear temp files...")
+    temp_dir.cleanup()
 
 def format_prompt(message, history, model):
     if model is None or 'mistral' in model.lower():
@@ -111,13 +118,17 @@ def bot(history, model=None):
     if model:
         local_llm = CustomLLM(repo_id=model, model_type=model_type, api_token=API_TOKEN)
 
-    if len(history) > 0 and len(history[-1]) > 0 and '.pdf' in history[-1][0]:
-        response = "You uploaded a PDF file. You can ask questions from the file."
-
-    elif len(history) > 0 and len(history[-1]) > 0 and '===>' in history[-1][0]:
+    
+    if len(history) > 0 and len(history[-1]) > 0 and '===>' in history[-1][0]:
         new_model = history[-1][0].replace("===>", "")
         response = f"You have changed the model to {new_model}"
 
+    elif len(history) > 0 and len(history[-1]) > 0 and '.pdf!!!' in history[-1][0].lower():
+        response = f"Unable to add file. Maximum 3 files allowed."
+
+    elif len(history) > 0 and len(history[-1]) > 0 and '.pdf' in history[-1][0]:
+        response = "You uploaded a PDF file. You can ask questions from the file."
+
     else:
         prompt = history[-1][0]
 
@@ -182,6 +193,7 @@ with gr.Blocks() as demo:
             container=False,
         )
         btn = gr.UploadButton("📁", file_types=[".pdf"])
+        clear_btn = gr.ClearButton([chatbot, txt])
 
     drop.change(modelChanged, [chatbot, drop], [chatbot, drop], queue=False).then(
         bot, [chatbot, drop], chatbot
@@ -196,6 +208,8 @@ with gr.Blocks() as demo:
         bot, [chatbot, drop], chatbot
     )
 
+    clear_btn.click(clearClick)
+
     
 
 demo.queue()