Update app.py

app.py

@@ -1,42 +1,22 @@
 import gradio as gr
-import requests
+from transformers import pipeline
 
-# -------------------------------
-# CONFIG
-# -------------------------------
-API_URL = "https://carecompanion-keywordextraction.onrender.com/extract_keywords"  # replace with your actual endpoint
+# Load the Hugging Face model for keyword/keyphrase extraction
+extractor = pipeline("text2text-generation", model="ml6team/keyphrase-extraction-distilbert-inspec")
 
-# -------------------------------
-# FUNCTION TO CALL RENDER API
-# -------------------------------
 def extract_keywords(text):
-    try:
-        payload = {"text": text}
-        response = requests.post(API_URL, json=payload)
+    if not text.strip():
+        return "⚠️ Please enter some text."
+    result = extractor(text, max_length=64, clean_up_tokenization_spaces=True)
+    return result[0]['generated_text']
 
-        if response.status_code == 200:
-            data = response.json()
-            # Customize this based on your API response format
-            return data.get("keywords", "No keywords found.")
-        else:
-            return f"⚠️ API Error {response.status_code}: {response.text}"
-    except Exception as e:
-        return f"❌ Connection error: {str(e)}"
-
-# -------------------------------
-# GRADIO INTERFACE
-# -------------------------------
 demo = gr.Interface(
     fn=extract_keywords,
-    inputs=gr.Textbox(label="Enter text for keyword extraction", placeholder="Paste your text here..."),
-    outputs=gr.Textbox(label="Extracted Keywords"),
-    title="🔍 Keyword Extraction API Demo",
-    description="This Gradio interface calls a deployed keyword extraction API hosted on Render.",
-    theme="soft"
+    inputs=gr.Textbox(lines=6, placeholder="Enter medical or general text here..."),
+    outputs="text",
+    title="🔍 Keyword Extraction",
+    description="Extract key phrases and keywords using a Hugging Face Transformer model."
 )
 
-# -------------------------------
-# LAUNCH
-# -------------------------------
 if __name__ == "__main__":
     demo.launch()