Update app.py

app.py

@@ -1,78 +1,67 @@
-import gradio as gr
-from instagrapi import Client
-import requests
-import tempfile
 import os
-import pathlib
+import tempfile
+import requests
+from fastapi import FastAPI, Form
+from instagrapi import Client
+from PIL import Image
 
-# ---------- Config ----------
-INSTAGRAM_USERNAME = os.getenv("IG_USERNAME")  # Hugging Face Secret
-INSTAGRAM_PASSWORD = os.getenv("IG_PASSWORD")  # Hugging Face Secret
-SESSION_FILE = "session.json"
+# Initialize FastAPI app
+app = FastAPI()
 
+# Instagram client
 cl = Client()
 
+# Load Instagram credentials from HF secrets
+IG_USERNAME = os.getenv("USERNAME")
+IG_PASSWORD = os.getenv("PASSWORD")
+
 def login():
-    """
-    Login with session file if available, otherwise with username/password.
-    """
     try:
-        if pathlib.Path(SESSION_FILE).exists():
-            cl.load_settings(SESSION_FILE)
-            cl.get_timeline_feed()  # test if session works
-            return "✅ Logged in with saved session"
-        else:
-            cl.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
-            cl.dump_settings(SESSION_FILE)
-            return "✅ Logged in with username/password"
+        cl.login(IG_USERNAME, IG_PASSWORD)
+        print("✅ Logged into Instagram")
     except Exception as e:
-        # If session fails, force new login
-        try:
-            cl.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
-            cl.dump_settings(SESSION_FILE)
-            return f"⚠️ Session failed, new login: {str(e)}"
-        except Exception as e2:
-            return f"❌ Login error: {str(e2)}"
+        print(f"⚠️ Login failed: {e}")
+        raise
 
-def upload_post(photo_url: str, caption: str):
-    """
-    Downloads photo and uploads to Instagram with caption.
-    """
+def fix_google_drive_url(url: str) -> str:
+    """Convert Google Drive share link to direct-download link"""
+    if "drive.google.com" in url:
+        if "/d/" in url:
+            file_id = url.split("/d/")[1].split("/")[0]
+            return f"https://drive.google.com/uc?export=download&id={file_id}"
+    return url
+
+@app.post("/upload")
+def upload_post(photo_url: str = Form(...), caption: str = Form(...)):
     try:
-        # Ensure login
         if not cl.user_id:
             login()
 
-        # Download photo
+        # Fix Google Drive links
+        photo_url = fix_google_drive_url(photo_url)
+
         response = requests.get(photo_url, stream=True)
         if response.status_code != 200:
-            return {"status": "error", "message": "Invalid image URL"}
+            return {"status": "error", "message": f"Invalid image URL: {response.status_code}"}
 
-        with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp_file:
-            for chunk in response.iter_content(1024):
-                tmp_file.write(chunk)
-            tmp_path = tmp_file.name
+        # Save downloaded file
+        tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
+        tmp_file.write(response.content)
+        tmp_file.close()
 
-        # Upload to Instagram
-        cl.photo_upload(tmp_path, caption)
+        # Validate image
+        try:
+            Image.open(tmp_file.name).verify()
+        except Exception:
+            return {"status": "error", "message": "Downloaded file is not a valid image."}
 
+        # Upload to Instagram
+        cl.photo_upload(tmp_file.name, caption)
         return {"status": "success", "message": "Photo uploaded successfully!"}
 
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
-# ---------- Gradio UI ----------
-with gr.Blocks() as demo:
-    gr.Markdown("## 📸 Instagram Auto Uploader (Hugging Face Space)")
-    with gr.Row():
-        photo_url = gr.Textbox(label="Photo URL")
-        caption = gr.Textbox(label="Caption")
-    output = gr.JSON(label="Result")
-    upload_btn = gr.Button("Upload to Instagram")
-
-    upload_btn.click(fn=upload_post, inputs=[photo_url, caption], outputs=output)
-
-# ---------- Launch ----------
-if __name__ == "__main__":
-    login()  # auto login on startup
-    demo.launch()
+@app.get("/")
+def root():
+    return {"status": "ok", "message": "Instagram uploader is running 🚀"}