Update app.py

app.py

@@ -233,9 +233,6 @@ def generate_chunked_response(model, prompt, max_tokens=1000, max_chunks=5):
             full_response += chunk
         except Exception as e:
             print(f"Error in generate_chunked_response: {e}")
-            if "Input validation error" in str(e):
-                # If we hit the token limit, return what we have so far
-                return full_response if full_response else "The input was too long to process. Please try a shorter query."
             break
     return full_response.strip()
 
@@ -258,7 +255,7 @@ _useragent_list = [
     "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Safari/537.36",
 ]
 
-def google_search(term, num_results=3, lang="en", timeout=5, safe="active", ssl_verify=None):
+def google_search(term, num_results=5, lang="en", timeout=5, safe="active", ssl_verify=None):
     escaped_term = urllib.parse.quote_plus(term)
     start = 0
     all_results = []
@@ -328,12 +325,6 @@ def google_search(term, num_results=3, lang="en", timeout=5, safe="active", ssl_
 
     return all_results
 
-def estimate_tokens(text):
-    # Rough estimate: 1 token ~= 4 characters
-    return len(text) // 4
-
-import re
-
 def estimate_tokens(text):
     # Rough estimate: 1 token ~= 4 characters
     return len(text) // 4
@@ -497,8 +488,7 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search, c
                 if attempt == max_attempts - 1:
                     return f"I apologize, but an unexpected error occurred. Please try again with a different question."
 
-    return "An unexpected error occurred. Please try again later."
-
+    return "An unexpected error occurred. Please try again later."."
 
 def extract_answer(full_response, instructions=None):
     answer_patterns = [
@@ -537,7 +527,6 @@ def extract_answer(full_response, instructions=None):
     
     return full_response.strip()
 
-
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Enhanced PDF Document Chat and Web Search")