Update app.py

app.py

@@ -4,7 +4,6 @@ from duckduckgo_search import DDGS
 import re
 
 client = InferenceClient("Pinkstack/Superthoughts-lite-v1")
-
 def format_search_results(query, results, result_type):
     formatted = f"{result_type} search results for '{query}':\n"
     for i, result in enumerate(results):
@@ -22,30 +21,6 @@ def extract_key_phrases(message):
             phrases.append(f"{words[i]} {words[i+1]}")
     return phrases[:3]
 
-def collect_sources(search_option, message):
-    sources = []
-    if search_option != "No search":
-        with DDGS() as ddgs:
-            if search_option == "Normal search":
-                web_results = ddgs.text(message, max_results=3)
-                sources.extend([(r.get('title', 'No title'), r.get('href', '') or r.get('url', '')) for r in web_results])
-            elif search_option == "Deep research":
-                queries = extract_key_phrases(message)
-                for query in queries:
-                    web_results = ddgs.text(query, max_results=3)
-                    news_results = ddgs.news(query, max_results=2)
-                    sources.extend([(r.get('title', 'No title'), r.get('href', '') or r.get('url', '')) for r in web_results])
-                    sources.extend([(r.get('title', 'No title'), r.get('url', '')) for r in news_results])
-    return sources
-
-def format_sources(sources):
-    if not sources:
-        return ""
-    formatted = "\n\n**Sources:**\n"
-    for i, (title, url) in enumerate(sources, 1):
-        formatted += f"{i}. [{title}]({url})\n"
-    return formatted
-
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -56,8 +31,7 @@ def respond(
     search_option,
 ):
     search_text = ""
-    sources = collect_sources(search_option, message)
-    if sources:
+    if search_option != "No search":
         with DDGS() as ddgs:
             if search_option == "Normal search":
                 web_results = ddgs.text(message, max_results=3)
@@ -93,12 +67,9 @@ def respond(
         if token:
             response += token
 
-    # Process response to convert tags to HTML divs and append sources
+    # Process response to convert tags to HTML divs
     formatted_response = response
     formatted_response = formatted_response.replace("<think>", '<div class="thinking">').replace("</think>", "</div>")
-    if search_option != "No search":
-        # Append sources to the output section
-        formatted_response = formatted_response.replace("</output>", f"{format_sources(sources)}</output>")
     formatted_response = formatted_response.replace("<output>", '<div class="output">').replace("</output>", "</div>")
 
     yield formatted_response
@@ -140,20 +111,26 @@ css = """
 
 system_message_default = """
 Respond in the following format:
+
 <think>
-...
+Your thought process here...
 </think>
+
 <output>
-...
+Your final answer here...
+
+If search results are provided, cite relevant sources at the end of this section as a numbered list in the format:
+1. [Title](URL)
+2. [Title](URL)
 </output>
 """
 
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Textbox(value=system_message_default, label="System message", lines=5),
-        gr.Slider(minimum=1, maximum=4096, value=1024, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=1.5, value=0.7, step=0.1, label="Temperature"),
+        gr.Textbox(value=system_message_default, label="System message", lines=8),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),
         gr.Dropdown(
             choices=["No search", "Normal search", "Deep research"],