Update app.py

app.py

@@ -277,7 +277,15 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
     print(buffer)
     print("---------")
 
-    return PlainTextResponse(buffer)
+        # If the generated text is JSON, parse it and format it as plain text
+    try:
+        json_content = json.loads(buffer)
+        formatted_text = "\n".join(f"{key}: {value}" for key, value in json_content.items())
+    except json.JSONDecodeError:
+        # If the buffer is not valid JSON, return it as is
+        formatted_text = buffer
+
+    return PlainTextResponse(formatted_text)