Update app.py

app.py

@@ -6,12 +6,33 @@ import urllib.parse
 import random
 import os
 from dotenv import load_dotenv
+import shutil
+import tempfile
 
 load_dotenv()  # Load environment variables from .env file
 
 # Now replace the hard-coded token with the environment variable
 HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
 
+def clear_cache():
+    try:
+        # Clear Gradio cache
+        cache_dir = tempfile.gettempdir()
+        shutil.rmtree(os.path.join(cache_dir, "gradio"), ignore_errors=True)
+        
+        # Clear any custom cache you might have
+        # For example, if you're caching PDF files or search results:
+        if os.path.exists("output_summary.pdf"):
+            os.remove("output_summary.pdf")
+        
+        # Add any other cache clearing operations here
+        
+        print("Cache cleared successfully.")
+        return "Cache cleared successfully."
+    except Exception as e:
+        print(f"Error clearing cache: {e}")
+        return f"Error clearing cache: {e}"
+
 _useragent_list = [
     "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
     "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
@@ -254,17 +275,20 @@ def scrape_and_display(query, num_results, instructions, web_search=True, temper
     return generated_summary
 
 # Main Gradio interface function
-def gradio_interface(query, use_pdf, pdf, num_results, instructions, temperature, repetition_penalty, top_p):
+def gradio_interface(query, use_pdf, pdf, num_results, instructions, temperature, repetition_penalty, top_p, clear_cache_flag):
+    if clear_cache_flag:
+        return clear_cache()
+    
     if use_pdf and pdf is not None:
         pdf_text = read_pdf(pdf)
         generated_summary = scrape_and_display(pdf_text, num_results=0, instructions=instructions, web_search=False, temperature=temperature, repetition_penalty=repetition_penalty, top_p=top_p)
     else:
         generated_summary = scrape_and_display(query, num_results=num_results, instructions=instructions, web_search=True, temperature=temperature, repetition_penalty=repetition_penalty, top_p=top_p)
-
+    
     # Save the generated summary to a PDF
     output_pdf_path = "output_summary.pdf"
     save_text_to_pdf(generated_summary, output_pdf_path)
-
+    
     return generated_summary, output_pdf_path
 
 # Deploy Gradio Interface
@@ -274,13 +298,15 @@ gr.Interface(
         gr.Textbox(label="Query"),
         gr.Checkbox(label="Use PDF"),
         gr.File(label="Upload PDF"),
-        gr.Slider(minimum=1, maximum=20, label="Number of Results"),  # Added Slider for num_results
+        gr.Slider(minimum=1, maximum=20, value=5, step=1, label="Number of Results"),
         gr.Textbox(label="Instructions"),
-        gr.Slider(minimum=0.1, maximum=1.0, label="Temperature"),
-        gr.Slider(minimum=0.1, maximum=1.0, label="Repetition Penalty"),
-        gr.Slider(minimum=0.1, maximum=1.0, label="Top p")
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Repetition Penalty"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.9, step=0.1, label="Top p"),
+        gr.Checkbox(label="Clear Cache", visible=False)  # Hidden checkbox for clear cache functionality
     ],
-    outputs=["text", "file"],  # Updated to return text and a file
+    outputs=["text", gr.File(label="Generated PDF")],
     title="Financial Analyst AI Assistant",
-    description="Enter your query about a company's financials to get valuable insights. Optionally, upload a PDF for analysis.Please instruct me for curating your output template, also for web search you can modify my search results but its advisable to restrict the same at 10. You can also adjust my parameters like Temperature, Repetition Penalty and Top_P, its adivsable to set repetition penalty at 1 and other two parameters at 0.1.",
+    description="Enter your query about a company's financials to get valuable insights. Optionally, upload a PDF for analysis. Please instruct me for curating your output template. For web search, you can modify my search results but it's advisable to restrict them to 10. You can also adjust parameters like Temperature, Repetition Penalty, and Top_P. It's advisable to set repetition penalty at 1 and other two parameters at 0.1.",
+    allow_flagging="never"
 ).launch(share=True)