Update app.py

app.py

@@ -14,11 +14,21 @@ headers = {"Authorization": f"Bearer {groq_api_key}"}
 
 # Function to interact with Groq API
 def chat_with_groq(user_input):
+    # Check if question is related to materials science
+    keywords = [
+        "material", "materials", "alloy", "composite", "polymer", "ceramic",
+        "application", "mechanical properties", "thermal properties", "corrosion",
+        "creep", "fatigue", "strength", "tensile", "impact", "fracture", "modulus"
+    ]
+    
+    if not any(word in user_input.lower() for word in keywords):
+        return "⚠️ I am an expert in Materials Science, ask me anything about it and I will try my best to answer. Anything outside, feel free to use ChatGPT! 🙂"
+
     system_prompt = (
         "You are an expert materials scientist. When a user asks about the best materials for a specific application, "
-        "provide the top 3 material choices. For each, include a brief justification with their relevant mechanical, "
-        "thermal, and chemical properties. First list the key properties required for that application, then present a comparison "
-        "table of the top 3 materials highlighting those properties. Format the response in markdown."
+        "provide the top 3 material choices. First, list the key properties required for that application. Then show a clean, "
+        "side-by-side comparison in markdown table format of the three materials, with the properties as rows and materials as columns. "
+        "Include their relevant mechanical, thermal, and chemical properties. Conclude with a brief summary of which might be best depending on the scenario."
     )
     
     body = {
@@ -32,19 +42,35 @@ def chat_with_groq(user_input):
     response = requests.post(url, headers=headers, json=body)
     
     if response.status_code == 200:
-        return response.json()['choices'][0]['message']['content']
+        base_response = response.json()['choices'][0]['message']['content']
+        
+        # Append popular related questions for UAE
+        popular_questions = (
+            "\n\n---\n"
+            "#### 🔍 Popular related questions in the UAE:\n"
+            "- What are the best corrosion-resistant materials for marine environments (e.g., desalination)?\n"
+            "- Which materials are ideal for solar panel coatings and desert heat management?\n"
+            "- What materials are used for aerospace structures in extreme climates?\n"
+            "- Best high-strength materials for construction in the Gulf region?\n"
+            "- What advanced materials are used in electric vehicles and batteries in the UAE?\n"
+        )
+        
+        return base_response + popular_questions
     else:
         return f"Error: {response.json()}"
 
-# Create Gradio interface
-interface = gr.Interface(
-    fn=chat_with_groq,
-    inputs=gr.Textbox(lines=2, placeholder="Ask about materials for a specific application..."),
-    outputs=gr.Markdown(),
-    title="Materials Science Expert Chatbot",
-    description="Ask about the best materials for a given application. Get the top 3 candidates with detailed comparison of their properties."
-)
+# Build improved layout using Gradio Blocks
+with gr.Blocks(title="Materials Science Expert Chatbot") as demo:
+    gr.Markdown("## 🧪 Materials Science Expert\nAsk about the best materials for a specific application.")
+    
+    with gr.Row():
+        user_input = gr.Textbox(lines=2, placeholder="e.g. Best materials for high-temperature turbine blades...", label="Ask your question")
+        submit_btn = gr.Button("Submit")
+    
+    output = gr.Markdown()
+    
+    submit_btn.click(chat_with_groq, inputs=user_input, outputs=output)
 
-# Launch Gradio app
+# Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()