Update app.py

app.py

@@ -2,23 +2,24 @@ import gradio as gr
 import os
 from groq import Groq
 
-# Fetch the Groq API key from Gradio secrets
+# Load API key securely from environment variable
 GROQ_API_KEY = os.getenv("Groq_API_Key")
 
-# Initialize Groq client with the secret API key
+# Initialize Groq client using the SDK
 client = Groq(api_key=GROQ_API_KEY)
 
-# System prompt to define bot behavior
+# System instruction for the bot
 SYSTEM_PROMPT = (
     "You are a Construction Material Advisor Bot. Your job is to recommend the best construction materials "
     "for civil engineering projects based on environmental conditions, budget, load requirements, and durability. "
-    "Be concise, informative, and explain the pros and cons of the recommended materials."
+    "Be concise, informative, and explain pros and cons of the materials."
 )
 
+# Model name
 MODEL_NAME = "llama3-70b-8192"
 
+# Chat function using Groq SDK
 def get_material_advice(user_query: str) -> str:
-    """Return a material recommendation generated by the Groq LLM."""
     try:
         chat_completion = client.chat.completions.create(
             model=MODEL_NAME,
@@ -29,15 +30,18 @@ def get_material_advice(user_query: str) -> str:
             temperature=0.7,
             max_tokens=500,
         )
-        return chat_completion.choices[0].message.content.strip()
-    except Exception as exc:
-        return f"Error: {exc}"
+        response = chat_completion.choices[0].message.content.strip()
+        # Convert plain text response to Markdown for better formatting
+        formatted_response = f"### 🧱 Material Recommendations\n\n{response}"
+        return formatted_response
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-# Gradio interface
+# Gradio UI
 demo = gr.Interface(
     fn=get_material_advice,
     inputs=gr.Textbox(lines=3, placeholder="Ask about construction materials..."),
-    outputs="text",
+    outputs=gr.Markdown(),
     title="🧱 Construction Material Advisor Bot",
     description=(
         "Ask questions like 'What is the best material for a coastal bridge?' or "
@@ -45,6 +49,5 @@ demo = gr.Interface(
     ),
 )
 
-# Launch the app
 if __name__ == "__main__":
     demo.launch()