Update app.py

app.py

@@ -3,87 +3,133 @@ import requests
 import gradio as gr
 from dotenv import load_dotenv
 
-# Load environment variables
+# Initialize environment
 load_dotenv()
 
-def summarize(text):
-    """Secure summarization with complete error handling"""
-    try:
-        # 1. Token verification
-        token = os.getenv("HF_API_TOKEN", "").strip()
-        if not token.startswith("hf_"):
-            return "🔒 Invalid token format - Check Secrets", "❌ Error"
+class Summarizer:
+    def __init__(self):
+        self.model_url = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
+        self.token = os.getenv("HF_API_TOKEN", "").strip()
         
-        # 2. Token validation API call
-        validation = requests.get(
-            "https://api-inference.huggingface.co/status",
-            headers={"Authorization": f"Bearer {token}"},
-            timeout=10
-        )
-        
-        if validation.status_code == 401:
-            return "🔑 Token rejected - Generate new READ token", "❌ Auth Failed"
-        
-        # 3. Main summarization
-        response = requests.post(
-            "https://api-inference.huggingface.co/models/facebook/bart-large-cnn",
-            headers={"Authorization": f"Bearer {token}"},
-            json={
-                "inputs": text[:5000],  # Character limit for safety
-                "parameters": {"max_length": 150}
-            },
-            timeout=30
-        )
-        
-        # 4. Handle all response cases
-        if response.status_code == 200:
-            summary = response.json()[0].get("summary_text", "No summary generated")
-            return summary, "✅ Success"
-        elif response.status_code == 503:
-            return "🔄 Model loading... Please wait 30 seconds", "⚠️ Retry"
-        else:
-            return f"API Error {response.status_code}", "❌ Failed"
+    def validate_token(self):
+        """Secure token verification with debug logging"""
+        if not self.token.startswith("hf_"):
+            print("❌ Token missing 'hf_' prefix")  # Debug to logs
+            return False
             
-    except requests.exceptions.RequestException as e:
-        return f"🌐 Network error: {str(e)}", "⚠️ Connection"
-    except Exception as e:
-        return f"Unexpected error: {str(e)}", "❌ Critical"
+        try:
+            test = requests.get(
+                "https://api-inference.huggingface.co/status",
+                headers={"Authorization": f"Bearer {self.token}"},
+                timeout=5
+            )
+            if test.status_code == 401:
+                print("🔑 Token rejected by API")  # Debug to logs
+            return test.ok
+        except:
+            return False
+
+    def summarize(self, text):
+        """Core summarization with robust error handling"""
+        try:
+            # Input validation
+            if not text.strip():
+                return "📝 Please enter text", "⚠️ Warning"
+                
+            # Token check
+            if not self.validate_token():
+                return "🔒 Invalid token - Check Secrets", "❌ Error"
+            
+            # API Call
+            response = requests.post(
+                self.model_url,
+                headers={"Authorization": f"Bearer {self.token}"},
+                json={
+                    "inputs": text[:5000],
+                    "parameters": {
+                        "max_length": 150,
+                        "min_length": 30,
+                        "do_sample": False
+                    }
+                },
+                timeout=25
+            )
+            
+            # Response handling
+            if response.status_code == 200:
+                result = response.json()[0].get("summary_text", "No summary generated")
+                return result, "✅ Success"
+            elif response.status_code == 503:
+                return "🔄 Model loading... Try again soon", "⚠️ Retry"
+            else:
+                return f"API Error {response.status_code}", "❌ Failed"
+                
+        except requests.exceptions.RequestException as e:
+            return f"🌐 Network issue: {str(e)}", "⚠️ Connection"
+        except Exception as e:
+            return f"Unexpected error: {str(e)}", "❌ Critical"
+
+# Initialize summarizer
+summarizer = Summarizer()
 
 # Gradio Interface
 with gr.Blocks(
     title="Summify Pro",
     theme=gr.themes.Soft(
-        primary_hue="blue",
-        font=[gr.themes.GoogleFont("Open Sans")]
-    )
+        primary_hue="indigo",
+        secondary_hue="blue",
+        font=[gr.themes.GoogleFont("Inter")]
+    ),
+    css=".gradio-container {max-width: 800px !important}"
 ) as app:
-    gr.Markdown("# 📝 Summify Pro")
+    gr.Markdown("""
+    # 📝 Summify Pro
+    *Secure text summarization powered by Hugging Face*
+    """)
+    
     with gr.Row():
         input_box = gr.Textbox(
-            label="Input Text",
-            placeholder="Paste or type your content here...",
-            lines=7,
-            max_lines=20
+            label="Your Text",
+            placeholder="Paste article, essay, or document...",
+            lines=8,
+            max_lines=20,
+            elem_id="input-box"
         )
         output_box = gr.Textbox(
-            label="Generated Summary",
+            label="Summary",
             interactive=False,
-            lines=7
+            lines=8,
+            elem_id="output-box"
         )
-    status = gr.Textbox(label="Status", interactive=False)
-    gr.Button(
+    
+    status = gr.Textbox(
+        label="System Status",
+        interactive=False,
+        elem_id="status-box"
+    )
+    
+    submit_btn = gr.Button(
         "Generate Summary",
-        variant="primary"
-    ).click(
-        fn=summarize,
+        variant="primary",
+        elem_id="submit-btn"
+    )
+    clear_btn = gr.Button("Clear")
+    
+    submit_btn.click(
+        fn=summarizer.summarize,
         inputs=input_box,
         outputs=[output_box, status]
     )
+    clear_btn.click(
+        lambda: ["", "", ""],
+        outputs=[input_box, output_box, status]
+    )
 
-# Production launch
+# Launch configuration
 if __name__ == "__main__":
     app.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_api=False
+        show_error=True,
+        debug=True
     )