fix: models

services/model_handler.py

@@ -101,6 +101,17 @@ class ModelHandler:
             add_references=True,
         )
     
+    def _format_prompt(self, role, instructions, query):
+        """Format the prompt for the model"""
+        return f"""Task: {role}
+
+Instructions:
+{instructions}
+
+Input: {query}
+
+Output:"""
+
     @staticmethod
     @st.cache_resource
     def _load_model():
@@ -120,7 +131,12 @@ class ModelHandler:
         if not api_key:
             raise ValueError("HuggingFace API key not found in environment variables")
         
-        return HuggingFace(id=MODEL_PATH, api_key=api_key)
+        return HuggingFace(
+            id=MODEL_PATH,
+            api_key=api_key,
+            task="text2text-generation",
+            max_length=512
+        )
 
     def _initialize_local_model(self):
         """Initialize local model as fallback"""
@@ -130,25 +146,33 @@ class ModelHandler:
     
     def generate_answer(self, query: str) -> str:
         try:
-            translator = self.translator.run(query, stream=False)
-            logging.info(f"Translated query")
-            research = self.researcher.run(translator.content, stream=False)
-            logging.info(f"Generated research")
-            summary = self.summarizer.run(research.content, stream=False)
-            logging.info(f"Generated summary")
-            presentation = self.presenter.run(summary.content, stream=False)
-            logging.info(f"Generated presentation")
+            # Format translation prompt
+            translation_prompt = self._format_prompt(
+                role="Translate the following text to English",
+                instructions="Provide a direct English translation of the input text.",
+                query=query
+            )
             
-            if not presentation.content:
-                return self._get_fallback_response()
-            return presentation.content
+            # Get English translation
+            translation = self.translator.run(translation_prompt, stream=False)
+            if not translation:
+                logging.error("Translation failed")
+                return "Error: Unable to translate the query"
+
+            # Format research prompt
+            research_prompt = self._format_prompt(
+                role="Research Assistant",
+                instructions="Provide a clear and concise answer based on scientific sources.",
+                query=translation
+            )
+            
+            # Get research results
+            research_results = self.researcher.run(research_prompt, stream=False)
+            if not research_results:
+                logging.error("Research failed")
+                return "Error: Unable to perform research"
+
+            return research_results
         except Exception as e:
             logging.error(f"Error generating answer: {str(e)}")
-            return self._get_fallback_response()
-    
-    @staticmethod
-    def _get_fallback_response() -> str:
-        """Provide a friendly, helpful fallback response"""
-        return """
-            Peço descula, mas encontrei um erro ao gerar a resposta. Tente novamente ou refaça a sua pergunta.
-        """
+            return f"Error: {str(e)}"