Debug

app.py

@@ -110,14 +110,23 @@ def chat_with_model(messages):
     pad_id = current_tokenizer.pad_token_id or current_tokenizer.unk_token_id or 0
     eos_id = current_tokenizer.eos_token_id
 
-    # --- Build system context
+        # Build system context
     system_messages = [
         {
             "role": "system",
             "content": (
                 "You are a radiologist's companion, here to answer questions about the patient and assist in the diagnosis if asked to do so. "
                 "You are able to call specialized tools. "
-                "At the moment, you have one tool available: an organ segmentation algorithm for abdominal CTs."
+                "At the moment, you have one tool available: an organ segmentation algorithm for abdominal CTs.\n\n"
+                "If the user requests an organ segmentation, output a JSON object in this structure:\n"
+                "{\n"
+                "  \"function\": \"segment_organ\",\n"
+                "  \"arguments\": {\n"
+                "    \"scan_path\": \"<path_to_ct_scan>\",\n"
+                "    \"organ\": \"<organ_name>\"\n"
+                "  }\n"
+                "}\n\n"
+                "Once you call the function, the app will execute it and return the result."
             )
         },
         {
@@ -130,7 +139,7 @@ def chat_with_model(messages):
     # filtered_messages = [msg for msg in messages if not (msg["role"] == "assistant" and "Welcome to the Radiologist's Companion" in msg["content"])]
 
     # FULL conversation
-    full_messages = system_messages + filtered_messages
+    full_messages = system_messages + messages
 
     # --- Generate from full context
     prompt = format_prompt(full_messages)