Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# ✅ Set up Hugging Face client
+# ✅ Set up Hugging Face client with Phi-3
 client = InferenceClient(
     model="microsoft/Phi-3-mini-4k-instruct",
     token=os.environ["HF_TOKEN"]
@@ -48,9 +48,14 @@ Return the result in this format:
 <code_here>
 ```
     """
-    # 🔄 Get text response
-    response = client.text_generation(prompt=prompt)
-    return response if response else "❌ Error: No response received."
+
+    response = client.chat_completion(
+        messages=[
+            {"role": "system", "content": "You are a helpful software architecture assistant."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message.content if response and response.choices else "❌ Error: No response received."
 
 # 🎨 Gradio UI
 with gr.Blocks() as demo:
@@ -104,4 +109,4 @@ with gr.Blocks() as demo:
         show_progress=True
     )
 
-    demo.launch()
+    demo.launch()