Upload app.py

app.py

@@ -3,11 +3,11 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# ✅ Correct IBM Granite model
+# ✅ IBM Granite model setup
 model_id = "ibm-granite/granite-3.3-2b-instruct"
-token = os.getenv("HF_TOKEN")  # Load from Hugging Face Secrets
+token = os.getenv("HF_TOKEN")  # Ensure your Hugging Face token is set in the environment
 
-# ✅ Load model and tokenizer using the updated syntax
+# ✅ Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=token)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
@@ -16,78 +16,52 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype=torch.float32
 )
 
+# ✅ Query function
 def query_granite(prompt):
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     outputs = model.generate(**inputs, max_new_tokens=100)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-# Pages
-def home():
-    with gr.Blocks() as demo:
-        gr.Markdown("# 🏥 Welcome to HealthAI")
-        gr.Markdown("Your intelligent healthcare assistant using IBM Granite 3.3-2B Instruct.")
-    return demo
+# ✅ Gradio UI using Tabs (instead of Pages)
+with gr.Blocks() as demo:
+    gr.Markdown("# 🏥 Welcome to HealthAI")
+    gr.Markdown("Your intelligent healthcare assistant using IBM Granite 3.3-2B Instruct.")
 
-def symptoms_app():
-    def identify(symptom):
-        return query_granite(f"What illness could cause: {symptom}?")
-    with gr.Blocks() as demo:
-        gr.Markdown("## 🩺 Symptom Identifier")
+    with gr.Tab("🩺 Symptoms"):
+        def identify(symptom):
+            return query_granite(f"What illness could cause: {symptom}?")
         symptom = gr.Textbox(label="Enter your symptom")
         output = gr.Textbox(label="AI Diagnosis")
         btn = gr.Button("Analyze")
         btn.click(identify, inputs=symptom, outputs=output)
-    return demo
 
-def remedies_app():
-    def get_remedies(issue):
-        return query_granite(f"What are home remedies for {issue}?")
-    with gr.Blocks() as demo:
-        gr.Markdown("## 🌿 Home Remedies")
+    with gr.Tab("🌿 Remedies"):
+        def get_remedies(issue):
+            return query_granite(f"What are home remedies for {issue}?")
         issue = gr.Textbox(label="What are you suffering from?")
-        output = gr.Textbox(label="Suggested Remedy")
-        btn = gr.Button("Suggest")
-        btn.click(get_remedies, inputs=issue, outputs=output)
-    return demo
+        remedy_output = gr.Textbox(label="Suggested Remedy")
+        remedy_btn = gr.Button("Suggest")
+        remedy_btn.click(get_remedies, inputs=issue, outputs=remedy_output)
 
-def diet_app():
-    def suggest(goal):
-        return query_granite(f"Suggest a diet for: {goal}")
-    with gr.Blocks() as demo:
-        gr.Markdown("## 🥗 Diet Suggestions")
+    with gr.Tab("🥗 Diet"):
+        def suggest(goal):
+            return query_granite(f"Suggest a diet for: {goal}")
         goal = gr.Textbox(label="Your health goal")
-        output = gr.Textbox(label="Diet Plan")
-        btn = gr.Button("Get Plan")
-        btn.click(suggest, inputs=goal, outputs=output)
-    return demo
+        diet_output = gr.Textbox(label="Diet Plan")
+        diet_btn = gr.Button("Get Plan")
+        diet_btn.click(suggest, inputs=goal, outputs=diet_output)
 
-def mental_app():
-    def tip(topic):
-        return query_granite(f"Mental health advice about: {topic}")
-    with gr.Blocks() as demo:
-        gr.Markdown("## 🧠 Mental Wellness")
+    with gr.Tab("🧠 Mental Wellness"):
+        def tip(topic):
+            return query_granite(f"Mental health advice about: {topic}")
         topic = gr.Textbox(label="Enter mental health topic")
-        output = gr.Textbox(label="Wellness Tip")
-        btn = gr.Button("Get Tip")
-        btn.click(tip, inputs=topic, outputs=output)
-    return demo
+        tip_output = gr.Textbox(label="Wellness Tip")
+        tip_btn = gr.Button("Get Tip")
+        tip_btn.click(tip, inputs=topic, outputs=tip_output)
 
-def faqs_app():
-    with gr.Blocks() as demo:
-        gr.Markdown("## ❓ FAQs")
+    with gr.Tab("❓ FAQs"):
+        gr.Markdown("### ❓ FAQs")
         gr.Markdown("**Q1:** What is HealthAI?")
         gr.Markdown("**A:** It's an AI assistant to help with health-related queries using IBM Granite 3.3-2B.")
-    return demo
 
-# App configuration
-pages = [
-    gr.Page(title="🏠 Home", path="/", block=home),
-    gr.Page(title="🩺 Symptoms", path="/symptoms", block=symptoms_app),
-    gr.Page(title="🌿 Remedies", path="/remedies", block=remedies_app),
-    gr.Page(title="🥗 Diet", path="/diet", block=diet_app),
-    gr.Page(title="🧠 Mental Wellness", path="/mental", block=mental_app),
-    gr.Page(title="❓ FAQs", path="/faqs", block=faqs_app),
-]
-
-app = gr.App(pages=pages)
-app.launch()
+demo.launch()