Update app.py

app.py

@@ -1,3 +1,4 @@
+from fastapi import FastAPI
 from utils.logging_utils import setup_logging
 from scripts.train_viral_potential import train_viral_potential
 from scripts.train_engagement_rate import train_engagement_rate
@@ -9,18 +10,27 @@ from scripts.analyze_engagement import analyze_engagement
 # Set up logging
 setup_logging()
 
-# Main application logic
-if __name__ == "__main__":
-    # Train models
+# Initialize FastAPI app
+app = FastAPI()
+
+@app.get("/")
+def read_root():
+    return {"message": "Instagram AI Backend"}
+
+@app.post("/train-models")
+def train_models():
     train_viral_potential()
     train_engagement_rate()
     train_promotion_strategy()
     train_time_series()
+    return {"message": "Models trained successfully"}
 
-    # Analyze engagement data
+@app.post("/analyze-engagement")
+def analyze_engagement_endpoint():
     analyze_engagement()
+    return {"message": "Engagement analysis completed"}
 
-    # Analyze an example image
-    image_url = "https://example.com/path/to/image.jpg"
-    caption = "This is a beautiful sunset!"
-    analyze_image_url(image_url, caption)
+@app.post("/analyze-image")
+def analyze_image_endpoint(image_url: str, caption: str):
+    analyze_image_url(image_url, caption)
+    return {"message": "Image analysis completed"}