Update app.py

app.py

@@ -1,86 +1,66 @@
 from flask import Flask, request, jsonify
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
-import torch
+from flask_cors import CORS  # Add this import
 import logging
-from pytrends.request import TrendReq
+import os
 
 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
+CORS(app)  # Add this to handle CORS issues
 
-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
-        )
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
-@app.route('/generate_script', methods=['POST'])
+@app.route('/generate_script', methods=['POST', 'GET'])  # Allow GET for testing
 def generate_script():
     try:
-        initialize_model()
+        logger.info("Received request to /generate_script")
+        
+        if request.method == 'GET':
+            return jsonify({"message": "Use POST method with {'topics': ['topic1', 'topic2']}"})
         
         data = request.json
+        logger.info(f"Request data: {data}")
+        
         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]}.
+        # For now, return a mock response to test the connection
+        mock_script = f"""VIRAL SCRIPT FOR: {topics[0]}
+
+Scene 1: Golden Hook
+- Visual: Eye-catching intro with text overlay "THIS WILL CHANGE EVERYTHING"
+- Voiceover: "What if I told you {topics[0].lower()} is about to revolutionize everything?"
 
-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
+Scene 2: Valuable Content
+- Visual: Demonstrating key points with graphics
+- Voiceover: "Here are 3 ways {topics[0].lower()} is changing the game..."
 
-SCRIPT STRUCTURE:
+Scene 3: CTA
+- Visual: Call-to-action screen with links
+- Voiceover: "Like and follow for more insights!"
 """
-        
-        # 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
+            "script": mock_script,
+            "full_topics": topics,
+            "status": "success"
         })
         
     except Exception as e:
-        logging.error(f"Error generating script: {str(e)}")
+        logger.error(f"Error: {str(e)}")
         return jsonify({"error": str(e)}), 500
 
 @app.route('/health', methods=['GET'])
 def health():
-    return jsonify({"status": "healthy"})
+    return jsonify({"status": "healthy", "message": "Server is running"})
+
+@app.route('/test', methods=['GET'])
+def test():
+    return jsonify({"message": "Test endpoint working!"})
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=False)
+    port = int(os.environ.get('PORT', 7860))
+    app.run(host='0.0.0.0', port=port, debug=False)