Create app.py

app.py

@@ -0,0 +1,86 @@
+from flask import Flask, request, jsonify
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+import torch
+import logging
+from pytrends.request import TrendReq
+
+app = Flask(__name__)
+logging.basicConfig(level=logging.INFO)
+
+# Initialize model (will load on first request)
+model_name = "google/gemma-1b-it"
+tokenizer = None
+model = None
+generator = None
+
+def initialize_model():
+    global tokenizer, model, generator
+    if tokenizer is None:
+        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        model = AutoModelForCausalLM.from_pretrained(
+            model_name,
+            torch_dtype=torch.float16,
+            device_map="auto"
+        )
+        generator = pipeline(
+            "text-generation",
+            model=model,
+            tokenizer=tokenizer,
+            device=0 if torch.cuda.is_available() else -1
+        )
+
+@app.route('/generate_script', methods=['POST'])
+def generate_script():
+    try:
+        initialize_model()
+        
+        data = request.json
+        topics = data.get('topics', [])
+        
+        if not topics:
+            return jsonify({"error": "No topics provided"}), 400
+        
+        # Create prompt for Gemma
+        prompt = f"""Create a viral 1-minute video script about {topics[0]}.
+
+REQUIREMENTS:
+- Duration: Exactly 1 minute
+- Include: Golden hook, valuable content, CTA
+- Virtual descriptions for each scene
+- Voiceover instructions for each scene
+- Make it engaging and viral-worthy
+
+SCRIPT STRUCTURE:
+"""
+        
+        # Generate script
+        result = generator(
+            prompt,
+            max_length=1024,
+            temperature=0.8,
+            do_sample=True,
+            top_p=0.9,
+            num_return_sequences=1
+        )
+        
+        generated_text = result[0]['generated_text']
+        
+        # Extract just the script part (remove the prompt)
+        script = generated_text.replace(prompt, "").strip()
+        
+        return jsonify({
+            "topic": topics[0],
+            "script": script,
+            "full_topics": topics
+        })
+        
+    except Exception as e:
+        logging.error(f"Error generating script: {str(e)}")
+        return jsonify({"error": str(e)}), 500
+
+@app.route('/health', methods=['GET'])
+def health():
+    return jsonify({"status": "healthy"})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=False)