Update app.py

app.py

@@ -156,13 +156,8 @@ class SimpleAgent:
             # Check if the question mentions an attached file
             has_attachment = "Attached file available at:" in question
             
-            # Create a much simpler prompt for better results
-            # Don't use the complex prompt structure that was causing issues
-            pass  # We'll create the prompt in the pipeline section
-            
-            logging.info(f"Sending prompt to model (first 200 chars): {prompt[:200]}...")
+            logging.info(f"Sending prompt to model (first 200 chars): Question: {question[:200]}...")
             logging.info(f"Full question length: {len(question)} characters")
-            logging.info(f"Full prompt length: {len(prompt)} characters")
             
             # Try different LLM approaches based on what's available
             try:
@@ -174,6 +169,7 @@ class SimpleAgent:
                     
                     # Create an even simpler, more direct prompt
                     simple_prompt = f"Q: {question[:200]}...\nA:"
+                    logging.info(f"Simple prompt length: {len(simple_prompt)} characters")
                     
                     outputs = self.text_generator(
                         simple_prompt,
@@ -221,8 +217,9 @@ class SimpleAgent:
                 else:
                     # InferenceClient approach
                     logging.info("Using InferenceClient...")
+                    basic_prompt = f"Question: {question}\nAnswer:"
                     response = self.client.text_generation(
-                        prompt,
+                        basic_prompt,
                         max_new_tokens=200,
                         temperature=0.3,
                         return_full_text=False