Update app.py

app.py

@@ -84,7 +84,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 @lru_cache(maxsize=100)
 def cached_search(query):
     try:
-        print(f"Performing search for: {query[:5000]}...")  
+        print(f"Performing search for: {query[:50000]}...")  
         result = search_tool(query)
         print(f"Search successful, returned {len(result)} characters")  
         return result
@@ -121,10 +121,10 @@ class BasicAgent:
         
         """Initialize the Gemini model with appropriate settings"""
         generation_config = {
-            "temperature": 0.7,
+            "temperature": 0.1,
             "top_p": 0.95,
             "top_k": 40,
-            "max_output_tokens": 1024,
+            "max_output_tokens": 10000,
         }
         
         safety_settings = {
@@ -218,13 +218,13 @@ class BasicAgent:
     
     def _extract_key_info(self, search_results, question):
         # Basic check for empty results
-        if not search_results or len(search_results) < 5:
+        if not search_results or len(search_results) < 15:
             return "No relevant information found."
             
         # For YouTube transcripts, extract the most relevant portion
         if "Transcript from YouTube video" in search_results:
             # Split by sentences but keep limited context
-            max_chars = 5000  # Keep a reasonable chunk size
+            max_chars = 10000  # Keep a reasonable chunk size
             if len(search_results) > max_chars:
                 # Take a portion from the middle of the transcript for better relevance
                 start_idx = search_results.find("\n") + 1  # Skip the first line which is the header
@@ -236,7 +236,7 @@ class BasicAgent:
         # Split results into sentences and find most relevant
         sentences = search_results.split('. ')
         if len(sentences) <= 100:
-            return search_results[:8000]  
+            return search_results[:10000]  
         
         # Try to find sentences with keywords from question
         keywords = [w for w in question.lower().split() if len(w) > 100]
@@ -269,19 +269,20 @@ class BasicAgent:
                 
             
                 prompt = f"""
-                Question: {question}
-    
-                Relevant information: {relevant_info}
-    
+                You: You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.            
                 Instructions:
                 1. Read the question and think about what you need to answer it. Stick to the question. If you need search results, use the search results. If not, just answer the question directly and ignore the search results. Do not use search results if you can answer the question without using them. 
                 2. If the question is not comprehensible, try reading each letter backwards, from the last character in the last word, to the first letter of the first word. Read carefully, all the way to the beginning. If the backwards text turns out to be an instruction, just follow the instruction and provide a direct answer. Don't provide comments. 
                 3. If the question is still not comprehensible, try seeing if it is in another language. 
-                4. Think about whether you need to elaborate on the information. For example, if you know that John and Jane are kids of Joan, you know Joan has at least two kids. In other words, if you don't have a number that is asked of you, see if you can count to produce an answer. Once you have counted, just answer the number. In the example I just gace, you would answer: "FINAL ANSWER: Two".
+                4. Think about whether you need to elaborate on the information. For example, if you know that John and Jane are kids of Joan, you know Joan has at least two kids. In other words, if you don't have a number that is asked of you, see if you can count to produce an answer. Once you have counted, just answer the number. Be succinct, coesive, I would even say tight in your answers. If the question asks "how many?", just reply back the number that answers. In the example I just gave, you would answer: "FINAL ANSWER: Two".
                 5. Provide a direct answer.
                 6. If the information doesn't contain the answer, say so honestly.
                 7. Do not invent anything. You can apply method to elaborate, but based on facts. Do not provide comments. Just the raw answer.  
                 8. Format your response as a direct answer. For example, if you are asked the year in which World War II began, just reply: "FINAL ANSWER: 1939".
+                
+                Question: {question}
+    
+                Relevant information: {relevant_info}
                 """
                 
                 response = self.gemini_model.generate_content(prompt)