Update app.py

app.py

@@ -167,25 +167,25 @@ class SimpleAgent:
                     # Use the transformers pipeline with optimized parameters
                     logging.info("Using pipeline approach...")
                     
-                    # Create a more factual, direct prompt
-                    if "mercedes" in question.lower() and "studio album" in question.lower():
-                        simple_prompt = f"Mercedes Sosa released how many studio albums between 2000-2009? Answer:"
-                    elif "teal'c" in question.lower() and "hot" in question.lower():
-                        simple_prompt = f"In Stargate, Teal'c responds to 'Isn't that hot?' by saying:"
-                    elif "vikings" in question.lower() and "where" in question.lower():
-                        simple_prompt = f"Where do the Minnesota Vikings play? Answer:"
-                    elif "chess" in question.lower() and "black" in question.lower():
-                        simple_prompt = f"Chess: Black's winning move in algebraic notation:"
-                    elif "how many" in question.lower():
-                        simple_prompt = f"Q: {question[:150]} A: The number is"
-                    elif "what is" in question.lower() or "who" in question.lower():
-                        simple_prompt = f"Q: {question[:150]} A:"
+                    # Create a general prompt that encourages direct, factual answers for any question
+                    if "how many" in question.lower():
+                        simple_prompt = f"{question} Answer with a number only:"
+                    elif "what is" in question.lower() or "what are" in question.lower():
+                        simple_prompt = f"{question} Answer briefly and factually:"
+                    elif "who" in question.lower():
+                        simple_prompt = f"{question} Answer with the name only:"
                     elif "where" in question.lower():
-                        simple_prompt = f"Q: {question[:150]} A: Location:"
+                        simple_prompt = f"{question} Answer with the location only:"
+                    elif "when" in question.lower():
+                        simple_prompt = f"{question} Answer with the date/time only:"
+                    elif "which" in question.lower():
+                        simple_prompt = f"{question} Answer with the specific item only:"
+                    elif question.strip().endswith("?"):
+                        simple_prompt = f"{question} Give a direct, factual answer:"
                     else:
-                        simple_prompt = f"Fact: {question[:150]} Answer:"
+                        simple_prompt = f"{question} Provide a brief, factual response:"
                     
-                    logging.info(f"Factual prompt: {simple_prompt[:100]}...")
+                    logging.info(f"General prompt pattern: {simple_prompt[:100]}...")
                     
                     outputs = self.text_generator(
                         simple_prompt,