improved prompt

cosmicforge_ai_chatbot/cosmicforge_chatbot.py

@@ -21,15 +21,14 @@ class CosmicForgeMedicalChat:
         await self.model.load_model()
 
     def create_medical_chat_prompt(self, user_input: str) -> str:
-        return f"""You are CosmicForge, an advanced medical AI assistant designed to provide comprehensive information on all aspects of health and medicine. Your knowledge spans general medical topics, health lifestyles, medical questions and answers, treatments, medications, and preventive care. Offer detailed, professional responses tailored to the user's query, always emphasizing the importance of consulting healthcare professionals for personalized medical advice, diagnosis, and treatment.
+        return f"""You are CosmicForge, an advanced medical AI assistant. Provide concise, professional responses to medical queries including:
+1. Key Information
+2. Brief Explanation
+3. Recommendations
 
 User Input: {user_input}
 
-If the input is related to health or medical topics:
-Provide a thorough, professional response addressing the user's query. Include relevant medical information, potential considerations, general advice, and always emphasize the importance of professional medical consultation.
-
-If the input is not related to medical or health topics, respond ONLY with:
-"I'm CosmicForge, a medical AI assistant. I can only provide information and advice related to health and medical topics. For other subjects, please consult appropriate resources or experts."
+If the input is not related to medical or health topics, respond ONLY with: "I'm CosmicForge, a medical AI assistant. I can only provide information on health and medical topics."
 """
 
     async def process_chat(self, user_input: str) -> Tuple[str, str]:
@@ -61,23 +60,38 @@ If the input is not related to medical or health topics, respond ONLY with:
         clean_response = response.replace(prompt, "").strip()
         lines = clean_response.split('\n')
         clean_lines = [line for line in lines if not line.startswith("User Input:") and "CosmicForge" not in line]
-        final_response = ' '.join(clean_lines).strip()
+        final_response = '\n'.join(clean_lines).strip()
         return final_response
 
     async def _post_process_response(self, response: str) -> str:
+        response = response.replace('Response:', '').replace('"', '').replace("'", "").strip()
+        first_section = "Key Information:"
+        if first_section in response:
+            response = response[response.index(first_section):]
+        sections = ["Key Information:", "Brief Explanation:", "Recommendations:"]
         clean_lines = []
+        current_section = ""
         for line in response.split('\n'):
             line = line.strip()
-            if line:
-                clean_lines.append(line)
+            if any(section in line for section in sections):
+                if clean_lines:
+                    clean_lines.append(" ")  # Single space between sections
+                current_section = line.strip(':')
+                clean_lines.append(current_section + ":")
+            elif line:
+                if current_section == "Key Information":
+                    clean_lines.append(line.strip('* '))
+                elif current_section == "Recommendations":
+                    clean_lines.append("- " + line.lstrip('0123456789. '))
+                else:
+                    clean_lines.append(line)
 
         clean_output = " ".join(clean_lines)
 
         if "I'm CosmicForge, a medical AI assistant." in clean_output:
             return clean_output
         
-        if "Important note:" not in clean_output:
-            clean_output += " Important note: This information is for educational purposes only and does not constitute medical advice. Please consult a healthcare professional for personalized medical guidance, diagnosis, and treatment."
+        clean_output += " Important note: This information is for educational purposes only and does not constitute medical advice. Please consult a healthcare professional for personalized medical guidance."
 
         return clean_output
 
@@ -132,4 +146,4 @@ If the input is not related to medical or health topics, respond ONLY with:
             except Exception as e:
                 logger.error(f"Unexpected error retrieving chat response status: {str(e)}", exc_info=True)
                 raise HTTPException(status_code=500, detail="Failed to retrieve chat response status")
-        return "not found"
+        return "not found"