Update app.py

app.py

@@ -271,15 +271,9 @@ def generate_chunked_response(prompt, model, max_tokens=10000, num_calls=3, temp
     print(f"Final clean response: {final_response[:100]}...")
     return final_response
 
-def duckduckgo_search(query,time=None):
-    params = {
-       "q": query,
-       "region": region,
-       "language": language,
-       "time": time,
-     }
+def duckduckgo_search(query):
     with DDGS() as ddgs:
-        results = ddgs.text(query, params=params, max_results=5)
+        results = ddgs.text(query, max_results=5)
     return results
 
 class CitingSources(BaseModel):
@@ -287,14 +281,14 @@ class CitingSources(BaseModel):
         ...,
         description="List of sources to cite. Should be an URL of the source."
     )
-def chatbot_interface(message, history, use_web_search, model, temperature, num_calls, region, language, time):
+def chatbot_interface(message, history, use_web_search, model, temperature, num_calls):
     if not message.strip():
         return "", history
 
     history = history + [(message, "")]
 
     try:
-        for response in respond(message, history, model, temperature, num_calls, use_web_search, region, language, time):
+        for response in respond(message, history, model, temperature, num_calls, use_web_search):
             history[-1] = (message, response)
             yield history
     except gr.CancelledError:
@@ -313,7 +307,7 @@ def retry_last_response(history, use_web_search, model, temperature, num_calls):
     
     return chatbot_interface(last_user_msg, history, use_web_search, model, temperature, num_calls)
 
-def respond(message, history, model, temperature, num_calls, use_web_search, region, language, time, selected_docs):
+def respond(message, history, model, temperature, num_calls, use_web_search, selected_docs):
     logging.info(f"User Query: {message}")
     logging.info(f"Model Used: {model}")
     logging.info(f"Search Type: {'Web Search' if use_web_search else 'PDF Search'}")
@@ -322,7 +316,7 @@ def respond(message, history, model, temperature, num_calls, use_web_search, reg
 
     try:
         if use_web_search:
-            for main_content, sources in get_response_with_search(message, model, num_calls=num_calls, temperature=temperature, region=region, language=language, time=time):
+            for main_content, sources in get_response_with_search(message, model, num_calls=num_calls, temperature=temperature):
                 response = f"{main_content}\n\n{sources}"
                 first_line = response.split('\n')[0] if response else ''
 #                logging.info(f"Generated Response (first line): {first_line}")
@@ -425,14 +419,16 @@ After writing the document, please provide a list of sources used in your respon
     if not full_response:
         yield "I apologize, but I couldn't generate a response at this time. Please try again later."
 
-def get_response_with_search(query, model, num_calls=3, temperature=0.2, region=region, language=language, time=time):
+def get_response_with_search(query, model, num_calls=3, temperature=0.2):
     search_results = duckduckgo_search(query)
     context = "\n".join(f"{result['title']}\n{result['body']}\nSource: {result['href']}\n" 
                         for result in search_results if 'body' in result)
     
     prompt = f"""Using the following context:
 {context}
-Write a detailed and complete research document that fulfills the following user request: '{query}'
+You are an expert AI assistant, write a detailed and complete research document that fulfills the following user request: '{query}'
+Base your entire response strictly on the information retrieved from trusted sources. Importantly, only include information that is directly supported by the retrieved content. 
+If any part of the information cannot be verified from the given sources, clearly state that it could not be confirmed.
 After writing the document, please provide a list of sources used in your response."""
 
     if model == "@cf/meta/llama-3.1-8b-instruct":
@@ -581,9 +577,6 @@ demo = gr.ChatInterface(
         gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[3]),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature"),
         gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls"),
-        gr.Textbox(label="Region", placeholder="US"),
-        gr.Textbox(label="Language", placeholder="en"),
-        gr.Textbox(label="Time", placeholder="d"),
         use_web_search,
         document_selector
     ],
@@ -667,4 +660,4 @@ with demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)