Update app.py

app.py

@@ -1,23 +1,37 @@
 import os
 import tempfile
 import requests
+import gradio as gr
 from fastapi import FastAPI, Form
 from instagrapi import Client
 from PIL import Image
 import uvicorn
 
+# FastAPI app (for API)
 app = FastAPI()
+
+# Instagram client
 cl = Client()
 
+# Instagram credentials from Hugging Face Secrets
 IG_USERNAME = os.getenv("USERNAME")
 IG_PASSWORD = os.getenv("PASSWORD")
 
+# Store logs in memory
+logs = []
+
+def log(message: str):
+    print(message)
+    logs.append(message)
+    if len(logs) > 50:  # keep last 50 logs
+        logs.pop(0)
+
 def login():
     try:
         cl.login(IG_USERNAME, IG_PASSWORD)
-        print("✅ Logged into Instagram")
+        log("✅ Logged into Instagram as " + IG_USERNAME)
     except Exception as e:
-        print(f"⚠️ Login failed: {e}")
+        log(f"⚠️ Login failed: {e}")
         raise
 
 def fix_google_drive_url(url: str) -> str:
@@ -27,17 +41,19 @@ def fix_google_drive_url(url: str) -> str:
             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(...)):
+def process_upload(photo_url: str, caption: str):
     try:
         if not cl.user_id:
             login()
 
         photo_url = fix_google_drive_url(photo_url)
+        log(f"📥 Downloading from {photo_url}")
 
         response = requests.get(photo_url, stream=True)
         if response.status_code != 200:
-            return {"status": "error", "message": f"Invalid image URL: {response.status_code}"}
+            msg = f"Invalid image URL: {response.status_code}"
+            log("❌ " + msg)
+            return {"status": "error", "message": msg}
 
         tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
         tmp_file.write(response.content)
@@ -46,18 +62,49 @@ def upload_post(photo_url: str = Form(...), caption: str = Form(...)):
         try:
             Image.open(tmp_file.name).verify()
         except Exception:
-            return {"status": "error", "message": "Downloaded file is not a valid image."}
+            msg = "Downloaded file is not a valid image."
+            log("❌ " + msg)
+            return {"status": "error", "message": msg}
 
         cl.photo_upload(tmp_file.name, caption)
+        log("✅ Photo uploaded successfully with caption: " + caption)
         return {"status": "success", "message": "Photo uploaded successfully!"}
 
     except Exception as e:
+        log("⚠️ Exception: " + str(e))
         return {"status": "error", "message": str(e)}
 
+# ----------------------------
+# FastAPI Endpoint (for Sheets)
+# ----------------------------
+@app.post("/upload")
+def upload_post(photo_url: str = Form(...), caption: str = Form(...)):
+    return process_upload(photo_url, caption)
+
 @app.get("/")
 def root():
     return {"status": "ok", "message": "Instagram uploader is running 🚀"}
 
-# 🔹 Add this so Hugging Face knows how to start it
+# ----------------------------
+# Gradio UI (for testing/debug)
+# ----------------------------
+def gradio_upload(photo_url, caption):
+    result = process_upload(photo_url, caption)
+    return result["status"], result["message"], "\n".join(logs)
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 📸 Instagram Auto Uploader")
+    with gr.Row():
+        photo_url_in = gr.Textbox(label="Photo URL", placeholder="Paste image link (Google Drive supported)")
+    caption_in = gr.Textbox(label="Caption", placeholder="Enter your caption")
+    upload_btn = gr.Button("🚀 Upload to Instagram")
+
+    status_out = gr.Textbox(label="Status")
+    message_out = gr.Textbox(label="Message")
+    logs_out = gr.Textbox(label="Logs (last 50)", lines=15)
+
+    upload_btn.click(fn=gradio_upload, inputs=[photo_url_in, caption_in], outputs=[status_out, message_out, logs_out])
+
+# 🔹 Run Gradio + FastAPI
 if __name__ == "__main__":
-    uvicorn.run("app:app", host="0.0.0.0", port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)