Update app.py

app.py

@@ -1,29 +1,42 @@
 import gradio as gr
-from transformers import pipeline
+import requests
 
-# ✅ Publicly available summarization model
-extractor = pipeline("summarization", model="facebook/bart-large-cnn")
+API_URL = "https://carecompanion-keywordextraction.onrender.com/extract_keywords"
 
-def extract_keywords(text):
+def extract_keywords_from_api(text):
     if not text.strip():
         return "⚠️ Please enter some text."
+    
     try:
-        result = extractor(text, max_length=40, min_length=5, do_sample=False)
-        keywords = result[0]['summary_text']
-        return f"🧠 **Extracted Keywords (Approx):**\n\n{keywords}"
+        response = requests.post(API_URL, json={"text": text})
+        if response.status_code != 200:
+            return f"❌ API Error: {response.status_code}"
+        
+        data = response.json()
+        
+        # Format output nicely
+        output = ""
+        for category, items in data.items():
+            if items:
+                output += f"### 🩸 {category.capitalize()}\n"
+                for item in items:
+                    if isinstance(item, list):
+                        output += f"- {item[0]} (score: {item[1]:.2f})\n"
+                    else:
+                        output += f"- {item}\n"
+                output += "\n"
+        
+        return output if output else "No keywords found."
+    
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"⚠️ Error: {str(e)}"
 
 demo = gr.Interface(
-    fn=extract_keywords,
-    inputs=gr.Textbox(
-        label="📝 Enter Text",
-        placeholder="Paste your text here...",
-        lines=6
-    ),
-    outputs=gr.Markdown(label="📊 Extracted Keywords"),
-    title="🔍 Keyword Extractor (using BART)",
-    description="A quick and lightweight keyword extractor built using `facebook/bart-large-cnn` summarization model.",
+    fn=extract_keywords_from_api,
+    inputs=gr.Textbox(label="Enter medical text to extract keywords:", lines=5, placeholder="Example: The patient is prescribed amoxicillin and paracetamol for fever."),
+    outputs=gr.Markdown(label="Extracted Keywords"),
+    title="🧠 Keyword Extraction from Medical Text",
+    description="Calls the Render API to extract structured medical keywords.",
     theme="soft"
 )