app.py

app.py

@@ -0,0 +1,62 @@
+# Step 1: Import the necessary toolkits
+import gradio as gr
+from transformers import pipeline
+
+# Step 2: Load our AI Model
+# This time, we are using a text-classification pipeline with our chosen health model.
+print("Loading the Health Analysis model...")
+health_classifier = pipeline(
+    "text-classification",
+    model="shanover/disease_classifier_base"
+)
+print("Model loaded successfully!")
+
+# Step 3: Define the main function for the app
+# This function takes the user's symptom text and returns a formatted prediction.
+def analyze_symptoms(symptoms):
+    # Use the pipeline to get the prediction. It returns a list of dictionaries.
+    predictions = health_classifier(symptoms)
+    # For this model, the top prediction is the first item in the list.
+    top_prediction = predictions[0]
+    
+    # Extract the predicted disease and the confidence score.
+    disease = top_prediction['label']
+    confidence_score = top_prediction['score']
+    
+    # Return the results in a clear, readable string.
+    return f"Predicted Condition: {disease}\nConfidence: {confidence_score:.2f}"
+
+# Step 4: Define the content for our app
+# We'll create a title, a detailed description, and a very important disclaimer.
+app_title = "Symptom to Disease Classifier"
+app_description = """
+Enter a list of symptoms, and this AI will predict a possible related medical condition.
+This tool is for educational purposes only and is based on the 'shanover/disease_classifier_base' model from Hugging Face.
+"""
+disclaimer = """
+**⚠️ IMPORTANT DISCLAIMER ⚠️**
+This is NOT a medical diagnostic tool. The predictions are generated by an AI model and may be inaccurate.
+**Always consult with a qualified healthcare professional for any medical advice or diagnosis.**
+This tool should not be used for any real-world medical decision-making.
+"""
+
+# Step 5: Create and Launch the Gradio Web App Interface
+app = gr.Interface(
+    fn=analyze_symptoms,
+    inputs=gr.Textbox(
+        lines=4,
+        placeholder="e.g., 'I am feeling vomiting, breathlessness, and sweating...'"
+    ),
+    outputs="text",
+    title=app_title,
+    description=app_description,
+    article=disclaimer,  # The 'article' parameter is great for adding extra text like our disclaimer.
+    examples=[
+        ["I have a persistent cough, high fever, and body aches."],
+        ["Feeling very thirsty and needing to urinate frequently."],
+        ["I have itchy skin with red patches and silver scales."]
+    ]
+)
+
+# Go live!
+app.launch()