app.py

app.py

@@ -0,0 +1,63 @@
+# app.py - NEW version to serve HTML and API
+from flask import Flask, request, jsonify, send_from_directory, render_template
+from transformers import pipeline
+import os
+
+# Configure Flask: 'templates' for HTML files, 'static' for CSS/JS (if external)
+app = Flask(__name__, template_folder='templates', static_folder='static')
+
+# --- Initialize the AI Model (t5-small) ---
+# This will download the model the first time your Space starts.
+try:
+    # device=-1 forces CPU, necessary for free tier and mobile environments.
+    generator = pipeline("text2text-generation", model="t5-small", device=-1)
+    print("AI Model 't5-small' loaded successfully!")
+except Exception as e:
+    print(f"Error loading AI model: {e}")
+    generator = None # Model failed to load
+
+# --- Route to Serve Your Main Website (index.html) ---
+@app.route('/')
+def serve_index():
+    # When someone visits your Space's main URL, this sends them your index.html
+    return render_template('index.html')
+
+# --- API Endpoint for AI Rewriting ---
+@app.route('/rewrite', methods=['POST'])
+def rewrite_text():
+    # Check if the AI model is ready
+    if generator is None:
+        return jsonify({"error": "AI model failed to load or is not ready."}), 503
+
+    # Get the message and tone from your website's request
+    data = request.get_json()
+    message = data.get('message', '')
+    tone = data.get('tone', 'kind') # Default to 'kind'
+
+    # Basic checks for valid input
+    if not message:
+        return jsonify({"error": "Message content is required."}), 400
+    if len(message) < 5:
+        return jsonify({"error": "Please provide a longer message for better AI rewriting."}), 400
+
+    # Craft the instruction for the AI
+    prompt = f"Rewrite the following message in a {tone} tone, maintaining its original meaning but adjusting the emotional nuance: '{message}'"
+
+    try:
+        # Generate rewritten text using the AI model
+        # max_length controls output length; num_beams improves quality but can be slower.
+        result = generator(prompt, max_length=150, min_length=20, num_beams=5, early_stopping=True)
+        rewritten_text = result[0]['generated_text']
+
+        # Send the rewritten text back to your website
+        return jsonify({"rewrittenText": rewritten_text})
+
+    except Exception as e:
+        print(f"Error during AI inference: {e}")
+        return jsonify({"error": f"An error occurred during AI rewriting: {str(e)}"}), 500
+
+# --- Main Part of the Flask Application ---
+if __name__ == '__main__':
+    # Hugging Face Spaces expects your app to run on port 7860
+    app.run(host='0.0.0.0', port=7860)
+