Update app.py

app.py

@@ -33,9 +33,9 @@ def summarize_text(text, model=DEFAULT_MODEL):
         load_dotenv()
         token = os.getenv("HF_API_TOKEN")
         if not token:
-            return "⚠️ Add your Hugging Face token to .env file"
+            return "⚠️ Add your Hugging Face token to .env file", "0/600"
     except Exception as e:
-        return f"⚠️ Config error: {str(e)}"
+        return f"⚠️ Config error: {str(e)}", "0/600"
 
     api_url = f"https://api-inference.huggingface.co/models/{model}"
     headers = {"Authorization": f"Bearer {token}"}
@@ -53,16 +53,16 @@ def summarize_text(text, model=DEFAULT_MODEL):
             if response.status_code == 200:
                 result = response.json()
                 if isinstance(result, list):
-                    return result[0].get("summary_text", "⚠️ No summary generated")
+                    return result[0].get("summary_text", "⚠️ No summary generated"), f"{quota['remaining']}/600"
             if response.status_code == 503:
                 time.sleep(RETRY_DELAY)
                 continue
-            return f"⚠️ API Error: {response.text[:200]}"
+            return f"⚠️ API Error: {response.text[:200]}", f"{quota['remaining']}/600"
         except requests.exceptions.RequestException as e:
             if attempt == MAX_RETRIES - 1:
-                return f"⚠️ Network error: {str(e)}"
+                return f"⚠️ Network error: {str(e)}", "0/600"
             time.sleep(RETRY_DELAY)
-    return "⚠️ Failed after multiple attempts"
+    return "⚠️ Failed after multiple attempts", "0/600"
 
 # --- Gradio Interface ---
 css = """
@@ -70,6 +70,8 @@ button {
     background: #6A0DAD !important;
     color: white !important;
     border: none !important;
+    border-radius: 8px !important;
+    padding: 12px 24px !important;
 }
 button:hover {
     background: #8A2BE2 !important;
@@ -82,14 +84,14 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, title="My AI Summarizer") as app
 
     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)
+        output_text = gr.Textbox(label="Summary", lines=8, interactive=False)
 
     with gr.Row():
         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), 
+        fn=summarize_text,
         inputs=input_text,
         outputs=[output_text, quota_display]
     )