Update app.py

app.py

@@ -142,32 +142,34 @@ class CodeAgent:
         )
 
     def create_prompt(self, question: str, search_results: Optional[List[Dict[str, str]]] = None) -> str:
-        """Create a prompt for the model with optional search results - Simplified Version"""
-        prompt_parts = []
-
-        if search_results:
-            prompt_parts.append(self.format_search_results(search_results))
-            prompt_parts.append(
-                "Instruction: Use the Search Results provided above to answer the Question. "
-                "Focus on extracting the specific fact, number, or list requested. "
-                "If the search results are clearly irrelevant or do not contain the answer, then try to use your general knowledge.\n\n"
-            )
-        else:
-            prompt_parts.append("Instruction: No search results were provided. Answer the Question using your general knowledge.\n\n")
-
-        prompt_parts.append(f"Question: {question}\n")
-        prompt_parts.append(
-            "\nAnswer Guidelines (Strictly Follow):\n"
-            "1. Your answer MUST be ONLY the specific fact, number(s), or comma-separated list requested by the question. NO extra words, explanations, or conversational filler like 'The answer is...'.\n"
-            "2. Example - Question: 'What is the capital of France?' Expected Answer: Paris\n"
-            "3. Example - Question: 'How many days are there in a standard week?' Expected Answer: 7\n"
-            "4. Example - Question: 'List the first three positive even numbers.' Expected Answer: 2, 4, 6\n"
-            "5. If, after carefully checking the search results (if any were provided) and your own knowledge, you absolutely cannot determine a factual answer, or if the question is impossible for you to answer (e.g. requires seeing an image), then respond with the single word: Unknown\n"
-            "   Do not guess. 'Unknown' is only for when you are certain you don't know.\n"
+    prompt = []
+
+    if search_results:
+        prompt.append("Search Results:\n")
+        for i, sr in enumerate(search_results, 1):
+            snippet = sr.get("snippet", "")
+            title = sr.get("title", "")
+            prompt.append(f"[{i}] Title: {title}\nSnippet: {snippet}\n")
+        prompt.append("\n")
+
+        prompt.append(
+            "Instructions:\n"
+            "- ONLY answer with the exact fact, number, name, or list from above search results. NO explanations, NO restating question.\n"
+            "- If several possible answers, pick the most obvious or first. Use comma for lists.\n"
+            "- Only use 'Unknown' if absolutely impossible to answer from the search results or common knowledge.\n"
+            "- If the answer needs audio/image/code execution, reply: Unknown.\n"
+            "\n"
         )
-        prompt_parts.append("\nAnswer: ")
-        
-        return "".join(prompt_parts)
+    else:
+        prompt.append(
+            "No search results. Use your knowledge. If impossible, reply: Unknown\n"
+        )
+    
+    prompt.append(f"Question: {question}\n")
+    prompt.append("Answer: ")
+    return "".join(prompt)
+
+
         
     def __call__(self, question: str) -> str:
         """