Update app.py

app.py

@@ -55,11 +55,17 @@ def get_medical_guidelines(wound_type):
         "HTTP-Referer": "https://huggingface.co/spaces/MahatirTusher/Wound_Treatment"
     }
     
-    prompt = f"""As a medical expert, provide treatment guidelines for {wound_type}:
-    - First aid steps
-    - Precautions
-    - When to seek professional help
-    Use clear, simple language without markdown."""
+    prompt = f"""You are a medical expert providing guidance for a patient with {wound_type}. 
+    First, briefly introduce what {wound_type} are and their typical characteristics.
+    Then, provide a comprehensive care guide including:
+
+    1. Immediate First Aid Steps
+    2. Home Care Instructions
+    3. Prevention Tips
+    4. Warning Signs (when to seek emergency care)
+    
+    Format your response in clear sections with proper spacing. Use simple, 
+    non-technical language that a general audience can understand."""
     
     try:
         print(f"📡 Sending API request for {wound_type}...")
@@ -113,16 +119,137 @@ def predict(image):
     except Exception as e:
         return {f"🚨 Error": str(e)}, ""
 
+# ================== CSS STYLES ==================
+custom_css = """
+    .gradio-container {
+        font-family: 'Times New Roman', Times, serif;
+    }
+    
+    .container {
+        background: rgba(255, 255, 255, 0.9);
+        border-radius: 20px;
+        padding: 2rem;
+        box-shadow: 0 8px 32px 0 rgba(31, 38, 135, 0.37);
+        backdrop-filter: blur(4px);
+        border: 1px solid rgba(255, 255, 255, 0.18);
+        margin: 1rem 0;
+    }
+    
+    .title {
+        color: #2D3748;
+        text-align: center;
+        font-size: 2.5rem;
+        margin-bottom: 1rem;
+    }
+    
+    .subtitle {
+        color: #4A5568;
+        text-align: center;
+        font-size: 1.2rem;
+        margin-bottom: 2rem;
+    }
+    
+    .warning {
+        background-color: #FED7D7;
+        border-left: 4px solid #F56565;
+        padding: 1rem;
+        margin: 1rem 0;
+        border-radius: 8px;
+    }
+    
+    .info-section {
+        background-color: #E6F6FF;
+        border-radius: 12px;
+        padding: 1.5rem;
+        margin: 1rem 0;
+    }
+"""
+
 # ================== GRADIO INTERFACE ==================
 def create_interface():
-    with gr.Blocks(title="AI Wound Classifier") as demo:
-        gr.Markdown("# 🩹 AI-Powered Wound Classification System")
-        gr.Markdown("Upload a wound image or take a photo using your camera")
+    with gr.Blocks(css=custom_css, title="WoundWise AI") as demo:
+        # Logo and Introduction
+        gr.Image("logo.png", show_label=False, container=False)
+        gr.Markdown("# Welcome to WoundWise AI")
+        
+        with gr.Box(class_name="container"):
+            gr.Markdown("""
+                In today's fast-paced world, early detection of medical conditions can be a game-changer. 
+                WoundWise AI is a cutting-edge AI-powered wound classification system developed as part 
+                of EarlyMed, an initiative by students of Vellore Institute of Technology. Our mission 
+                is simple yet impactful: "Early Detection, Smarter Decision."
+
+                By leveraging deep learning, this system analyzes wound images to provide accurate 
+                classifications and essential treatment guidelines. Whether it's a minor cut or a 
+                serious skin condition, our AI assists in identifying potential risks, helping users 
+                make informed health decisions at an early stage.
+            """)
+
+        # Main Interface
+        with gr.Box(class_name="container"):
+            gr.Markdown("## 📸 Upload or Capture Wound Image")
+            gr.Markdown("Use your device's camera or upload an existing image")
+            
+            file_input = gr.Image(type="pil", label="Upload Wound Image")
+            submit_btn = gr.Button("Analyze Now", variant="primary")
+            
+            # Example Images
+            gr.Examples(
+                examples=["abrasion.jpg", "burn.png", "bruise.png", "chicken-pox.png", "cut.png"],
+                inputs=file_input,
+                label="Example Images"
+            )
+            
+            output_label = gr.Label(label="Analysis Results", num_top_classes=3)
+            output_guidelines = gr.Textbox(
+                label="Medical Guidelines",
+                lines=12,
+                elem_classes="guidelines-box"
+            )
+
+        # Why Care About Wounds
+        with gr.Box(class_name="info-section"):
+            gr.Markdown("""
+                ## ⚠️ Why Wound Care Matters
+                
+                Ignoring wounds can lead to serious complications:
+                - Risk of bacterial infection
+                - Delayed healing and scarring
+                - Chronic wound development
+                - Systemic health issues
+                - Increased medical costs
+                
+                Early intervention and proper care are crucial for optimal healing and preventing complications.
+            """)
+
+        # How Our Model Works
+        with gr.Box(class_name="info-section"):
+            gr.Markdown("""
+                ## 🔬 Our Technology
+                
+                WoundWise AI uses a state-of-the-art deep learning model based on the GoogLeNet 
+                architecture, trained on thousands of medical images. The system:
+                
+                1. Analyzes wound images using advanced computer vision
+                2. Identifies 18 different types of wounds and skin conditions
+                3. Provides real-time classification with confidence scores
+                4. Generates medical guidelines using advanced AI
+                
+                Our model achieves high accuracy through extensive training and validation by 
+                medical professionals.
+            """)
 
-        file_input = gr.Image(type="pil", label="Upload Wound Image")
-        submit_btn = gr.Button("Analyze Now", variant="primary")
-        output_label = gr.Label(label="Top Predictions", num_top_classes=3)
-        output_guidelines = gr.Textbox(label="Treatment Guidelines", lines=8)
+        # Disclaimer
+        with gr.Box(class_name="warning"):
+            gr.Markdown("""
+                ## ⚕️ Medical Disclaimer
+                
+                This AI system is designed to assist in preliminary wound assessment only. It is not 
+                a substitute for professional medical advice, diagnosis, or treatment. Always seek 
+                the advice of qualified healthcare providers with any questions regarding your 
+                medical condition. If you believe your wound is serious or life-threatening, 
+                please seek immediate medical attention.
+            """)
 
         # Connect input to processing
         submit_btn.click(
@@ -138,5 +265,5 @@ if __name__ == "__main__":
     iface.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True  # Set to False if you do not want a public link
-    )
+        share=True
+    )