Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import json
 import logging
 from datetime import datetime
@@ -11,7 +10,6 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # System prompts remain the same as before
-# System prompts
 CONVERSATION_PROMPT = """You are LOSS DOG, a professional profile builder. Your goal is to have natural conversations
 with users to gather information about their professional background across 9 categories:
 
@@ -29,7 +27,8 @@ Be friendly and conversational. Ask follow-up questions naturally. When appropri
 but respect their boundaries. Once you believe you have gathered sufficient information (or if the user indicates they
 have nothing more to share), let them know they can click 'Generate Profile' to proceed.
 """
-EXTRACTION_PROMPT = """You are a professional information extraction system. Your task is to methodically analyze conversations and organize information into 9 specific categories. Process each category thoroughly and output in structured JSON format.
+
+EXTRACTION_PROMPT = """You are a professional information extraction system. Your task is to methodically analyze conversations and organize information into 9 specific categories. Process each category thoroughly and output in structured JSON format with no additional text.
 
 ANALYTICAL PROCESS:
 1. Read entire conversation history
@@ -208,7 +207,9 @@ Remember to:
 - Cross-reference information for consistency
 - Make reasonable inferences when appropriate
 - Maintain consistent formatting
-- Include all required fields even if empty"""
+- Include all required fields even if empty
+"""
+
 class ProfileBuilder:
     def __init__(self):
         self.conversation_history = []
@@ -270,12 +271,21 @@ class ProfileBuilder:
                 temperature=0.3
             )
 
-            # Parse the structured response
-            profile_data = json.loads(completion.choices[0].message.content)
+            # Get the raw output text from the model
+            raw_output = completion.choices[0].message.content
+            logger.info(f"Raw extraction output: {raw_output}")
+
+            # Attempt to parse the JSON
+            try:
+                profile_data = json.loads(raw_output)
+            except json.JSONDecodeError as decode_error:
+                logger.error("Failed to decode JSON. The output may not be valid JSON.")
+                profile_data = None  # Indicate failure to parse
 
-            # Add metadata
+            # Build the profile output including metadata and raw output
             profile = {
                 "profile_data": profile_data,
+                "raw_output": raw_output,
                 "metadata": {
                     "generated_at": datetime.now().isoformat(),
                     "conversation_length": len(self.conversation_history)
@@ -323,7 +333,10 @@ def create_gradio_interface():
 
             with gr.Column(scale=1):
                 generate_btn = gr.Button("Generate Profile")
-                profile_output = gr.JSON(label="Generated Profile")
+                # JSON output for structured profile
+                profile_output = gr.JSON(label="Generated Profile (Parsed JSON)")
+                # Markdown output to always show the raw AI output
+                raw_output_markdown = gr.Markdown(label="Raw Output from AI")
                 download_btn = gr.File(label="Download Profile")
 
         # Event handlers
@@ -342,8 +355,12 @@ def create_gradio_interface():
         async def on_generate():
             result = await builder.generate_profile()
             if "error" in result:
-                return {"error": result["error"]}, None
-            return result["profile"], result["filename"]
+                error_text = f"Error generating profile: {result['error']}"
+                return {"error": error_text}, None, error_text
+            profile = result["profile"]
+            # Prepare the raw output as markdown. Wrapping in triple backticks for code formatting.
+            raw_markdown = f"```json\n{profile.get('raw_output', '')}\n```"
+            return profile, result["filename"], raw_markdown
 
         # Bind events
         msg.submit(
@@ -360,7 +377,7 @@ def create_gradio_interface():
         
         generate_btn.click(
             on_generate,
-            outputs=[profile_output, download_btn]
+            outputs=[profile_output, download_btn, raw_output_markdown]
         )
 
     return demo
@@ -371,4 +388,4 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860
-    )
+    )