Update app.py

app.py

@@ -1,41 +1,107 @@
 import os
+import time
+import json
+from pathlib import Path
+from dotenv import load_dotenv
 import requests
+from datetime import datetime
 import gradio as gr
 
-TOKEN = os.getenv("HF_API_TOKEN", "").strip()
+# Configuration
+QUOTA_FILE = "api_quota.json"
+RATE_LIMIT_DELAY = 1.0
+DEFAULT_MODEL = "facebook/bart-large-cnn"
+MAX_RETRIES = 3
+RETRY_DELAY = 5
 
-def summarize(text):
-    if not text.strip(): return "Please enter text", ""
+# --- Core Functions ---
+def load_quota():
     try:
-        response = requests.post(
-            "https://api-inference.huggingface.co/models/facebook/bart-large-cnn",
-            headers={"Authorization": f"Bearer {TOKEN}"},
-            json={"inputs": text[:2000], "parameters": {"max_length": 100}},
-            timeout=20
-        )
-        return (response.json()[0]['summary_text'], "✅ Success") if response.ok else (f"⚠️ Error {response.status_code}", "")
+        if Path(QUOTA_FILE).exists():
+            with open(QUOTA_FILE, "r") as f:
+                return json.load(f)
+    except:
+        pass
+    return {"remaining": 600, "last_updated": datetime.now().isoformat()}
+
+def save_quota(quota):
+    with open(QUOTA_FILE, "w") as f:
+        json.dump(quota, f)
+
+def summarize_text(text, model=DEFAULT_MODEL):
+    try:
+        load_dotenv()
+        token = os.getenv("HF_API_TOKEN")
+        if not token:
+            return "⚠️ Add your Hugging Face token to .env file"
     except Exception as e:
-        return f"🚨 {str(e)}", ""
+        return f"⚠️ Config error: {str(e)}"
+
+    api_url = f"https://api-inference.huggingface.co/models/{model}"
+    headers = {"Authorization": f"Bearer {token}"}
+    payload = {"inputs": text, "parameters": {"max_length": 50, "min_length": 20}}
+
+    for attempt in range(MAX_RETRIES):
+        try:
+            time.sleep(RATE_LIMIT_DELAY)
+            response = requests.post(api_url, headers=headers, json=payload, timeout=10)
+            
+            quota = load_quota()
+            quota["remaining"] = max(0, quota.get("remaining", 600) - 1)
+            save_quota(quota)
+
+            if response.status_code == 200:
+                result = response.json()
+                if isinstance(result, list):
+                    return result[0].get("summary_text", "⚠️ No summary generated")
+            if response.status_code == 503:
+                time.sleep(RETRY_DELAY)
+                continue
+            return f"⚠️ API Error: {response.text[:200]}"
+        except requests.exceptions.RequestException as e:
+            if attempt == MAX_RETRIES - 1:
+                return f"⚠️ Network error: {str(e)}"
+            time.sleep(RETRY_DELAY)
+    return "⚠️ Failed after multiple attempts"
 
+# --- Gradio Interface ---
 css = """
 button {
     background: #6A0DAD !important;
     color: white !important;
     border: none !important;
-    border-radius: 8px !important;
 }
 button:hover {
     background: #8A2BE2 !important;
 }
 """
 
-with gr.Blocks(css=css) as app:
-    gr.Markdown("## ✨ SUMMIFY PRO")
+with gr.Blocks(theme=gr.themes.Soft(), css=css, title="My AI Summarizer") as app:
+    gr.Markdown("# ✨ My Text Summarizer")
+    gr.Markdown("Paste any text and get a concise summary using AI!")
+
+    with gr.Row():
+        input_text = gr.Textbox(label="Input Text", lines=8, placeholder="Paste your article/story here...")
+        output_text = gr.Textbox(label="Summary", lines=8)
+
     with gr.Row():
-        input_txt = gr.Textbox(label="Your Text", placeholder="Type here...", lines=5)
-        output_txt = gr.Textbox(label="Summary", interactive=False, lines=5)
-    status = gr.Textbox(label="Status", interactive=False)
-    gr.Button("Summarize").click(summarize, input_txt, [output_txt, status])
-    gr.Button("Clear").click(lambda: ["", "", ""], [input_txt, output_txt, status])
+        submit_btn = gr.Button("Summarize", variant="primary")
+        quota_display = gr.Textbox(label="API Status", interactive=False)
+
+    submit_btn.click(
+        fn=lambda x: (summarize_text(x), 
+        inputs=input_text,
+        outputs=[output_text, quota_display]
+    )
+
+    gr.Examples(
+        examples=[
+            "He prayed Fajr half-asleep...",
+            "Scientists discovered octopuses can edit their RNA...",
+            "The quick brown fox jumps over the lazy dog."
+        ],
+        inputs=input_text
+    )
 
-app.launch(server_name="0.0.0.0")
+if __name__ == "__main__":
+    app.launch()