Create app.py

app.py

@@ -1,63 +0,0 @@
-# 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)
-