Update app.py

app.py

@@ -3,67 +3,66 @@ from dotenv import load_dotenv
 from huggingface_hub import InferenceClient
 import os
 
-# Load env variables
 load_dotenv()
 
-# Initialize client
+app = Flask(__name__)
+
 client = InferenceClient(
-    provider="hf-inference",
     api_key=os.getenv("HF_API_KEY")
 )
 
-app = Flask(__name__)
-
-# Models to try in order (first working one wins)
 MODELS = [
     "mistralai/Mistral-7B-Instruct-v0.3",
     "HuggingFaceH4/zephyr-7b-beta",
-    "google/flan-t5-large",
 ]
 
-def generate_blog(topic, mode="creative", language="English"):
-    tone = "creative and engaging" if mode == "creative" else "professional and formal"
-    prompt = (
-        f"Write a well-structured, {tone} blog paragraph in {language} "
-        f"about: {topic}\n\nBlog:"
-    )
+def generate_blog(topic):
+    prompt = f"""
+Write a professional and engaging blog paragraph about:
+{topic}
+
+Blog:
+"""
 
     for model in MODELS:
         try:
             response = client.text_generation(
                 model=model,
                 prompt=prompt,
-                max_new_tokens=300,
+                max_new_tokens=250,
                 temperature=0.7,
             )
-            return response.strip()
-        except Exception as e:
-            err = str(e)
-            # Only fall through on "not supported" / quota errors
-            if "not supported" in err or "quota" in err.lower() or "rate" in err.lower():
-                continue
-            # Any other error — return immediately with message
-            return f"Error generating blog: {err}"
 
-    return "Error: All models are currently unavailable. Please try again later."
+            return str(response).strip()
 
+        except Exception as e:
+            print(f"Model failed: {model}")
+            print(e)
+            continue
+
+    return "Error: Could not generate blog."
 
 @app.route("/", methods=["GET", "POST"])
 def index():
     paragraph = ""
     error = ""
+
     if request.method == "POST":
-        topic    = request.form.get("topic", "").strip()
-        mode     = request.form.get("mode", "creative")
-        language = request.form.get("language", "English")
+        topic = request.form.get("topic", "").strip()
+
         if topic:
-            result = generate_blog(topic, mode, language)
+            result = generate_blog(topic)
+
             if result.startswith("Error"):
                 error = result
             else:
                 paragraph = result
-    return render_template("index.html", paragraph=paragraph, error=error)
 
+    return render_template(
+        "index.html",
+        paragraph=paragraph,
+        error=error
+    )
 
 if __name__ == "__main__":
-    app.run(debug=True, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+    app.run(debug=True)