Upload 3 files

Dockerfile

@@ -1,16 +1,16 @@
 FROM python:3.11-slim
 
-ENV PYTHONUNBUFFERED=1 \
-    PYTHONDONTWRITEBYTECODE=1
-
 WORKDIR /app
 
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-COPY main.py .
+COPY bot.py .
 
-# Writable dir for the Telethon session file (HF Spaces containers run as non-root by default)
-RUN mkdir -p /app/data && chmod -R 777 /app
+# ملاحظة: مو لازم فتح بورت لأن هذا بوت polling (MTProto)
+# مو ويب سيرفر، لكن نضيف واحد بسيط بس عشان Hugging Face
+# يتأكد إن الـ Space "شغال" (health check).
+ENV PORT=7860
+EXPOSE 7860
 
-CMD ["python", "main.py"]
+CMD ["python", "bot.py"]

bot.py

@@ -0,0 +1,225 @@
+import os
+import io
+import json
+import base64
+import logging
+from datetime import date
+
+import requests
+from pyrogram import Client, filters
+from pyrogram.types import Message
+
+logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
+log = logging.getLogger("image-bot")
+
+# ---------------------------------------------------------------------------
+# الإعدادات (كلها تُقرأ من متغيرات البيئة / HF Space secrets)
+# ---------------------------------------------------------------------------
+API_ID = int(os.environ["API_ID"])          # من my.telegram.org
+API_HASH = os.environ["API_HASH"]           # من my.telegram.org
+BOT_TOKEN = os.environ["BOT_TOKEN"]         # من BotFather
+GEMINI_API_KEY = os.environ["GEMINI_API_KEY"]
+# آيدي مجموعة وحدة أو أكثر مفصولة بفاصلة، مثال: "-1001111111111,-1002222222222"
+GROUP_IDS = [int(x.strip()) for x in os.environ["GROUP_IDS"].split(",") if x.strip()]
+
+DAILY_LIMIT = 2
+GEMINI_MODEL = "gemini-2.5-flash-image"
+GEMINI_URL = (
+    f"https://generativelanguage.googleapis.com/v1beta/models/"
+    f"{GEMINI_MODEL}:generateContent?key={GEMINI_API_KEY}"
+)
+
+COUNTS_FILE = "counts.json"
+
+app = Client(
+    "image_bot",
+    api_id=API_ID,
+    api_hash=API_HASH,
+    bot_token=BOT_TOKEN,
+)
+
+# ---------------------------------------------------------------------------
+# تخزين بسيط للعداد اليومي لكل مستخدم (يُحفظ بملف JSON)
+# ملاحظة: تخزين الـ Space على Hugging Face مؤقت، إذا انعاد تشغيل الـ Space
+# تنصفر العدادات. لو تحتاج ثبات دائم استخدم قاعدة بيانات خارجية لاحقاً.
+# ---------------------------------------------------------------------------
+def _load_counts() -> dict:
+    if os.path.exists(COUNTS_FILE):
+        try:
+            with open(COUNTS_FILE, "r", encoding="utf-8") as f:
+                return json.load(f)
+        except Exception:
+            return {}
+    return {}
+
+
+def _save_counts(data: dict) -> None:
+    with open(COUNTS_FILE, "w", encoding="utf-8") as f:
+        json.dump(data, f)
+
+
+def check_and_increment(user_id: int) -> bool:
+    """يرجع True إذا يقدر يسوي صورة، ويزيد العداد. يرجع False إذا وصل الحد."""
+    counts = _load_counts()
+    key = str(user_id)
+    today = date.today().isoformat()
+    entry = counts.get(key)
+
+    if not entry or entry.get("date") != today:
+        entry = {"date": today, "count": 0}
+
+    if entry["count"] >= DAILY_LIMIT:
+        counts[key] = entry
+        _save_counts(counts)
+        return False
+
+    entry["count"] += 1
+    counts[key] = entry
+    _save_counts(counts)
+    return True
+
+
+def remaining_today(user_id: int) -> int:
+    counts = _load_counts()
+    entry = counts.get(str(user_id))
+    today = date.today().isoformat()
+    if not entry or entry.get("date") != today:
+        return DAILY_LIMIT
+    return max(0, DAILY_LIMIT - entry["count"])
+
+
+# ---------------------------------------------------------------------------
+# استدعاء Gemini API
+# ---------------------------------------------------------------------------
+def gemini_generate(prompt: str) -> bytes | None:
+    """توليد صورة جديدة من نص فقط."""
+    body = {
+        "contents": [{"parts": [{"text": prompt}]}],
+        "generationConfig": {"responseModalities": ["IMAGE"]},
+    }
+    return _call_gemini(body)
+
+
+def gemini_edit(image_bytes: bytes, prompt: str, mime_type: str = "image/jpeg") -> bytes | None:
+    """تعديل صورة موجودة حسب وصف نصي."""
+    b64 = base64.b64encode(image_bytes).decode("utf-8")
+    body = {
+        "contents": [
+            {
+                "parts": [
+                    {"inline_data": {"mime_type": mime_type, "data": b64}},
+                    {"text": prompt},
+                ]
+            }
+        ],
+        "generationConfig": {"responseModalities": ["IMAGE"]},
+    }
+    return _call_gemini(body)
+
+
+def _call_gemini(body: dict) -> bytes | None:
+    try:
+        resp = requests.post(GEMINI_URL, json=body, timeout=120)
+        resp.raise_for_status()
+        data = resp.json()
+    except Exception as e:
+        log.error(f"Gemini API error: {e}")
+        return None
+
+    try:
+        parts = data["candidates"][0]["content"]["parts"]
+        for part in parts:
+            inline = part.get("inlineData") or part.get("inline_data")
+            if inline and inline.get("data"):
+                return base64.b64decode(inline["data"])
+    except Exception as e:
+        log.error(f"Gemini response parse error: {e} | raw: {data}")
+    return None
+
+
+# ---------------------------------------------------------------------------
+# فلتر: فقط داخل المجموعة المحددة
+# ---------------------------------------------------------------------------
+in_allowed_group = filters.chat(GROUP_IDS)
+
+
+@app.on_message(filters.command("start") & in_allowed_group)
+async def start_cmd(_, message: Message):
+    await message.reply_text(
+        "أهلاً 👋\n"
+        f"- /image <وصف> لتوليد صورة جديدة (بحد أقصى {DAILY_LIMIT} باليوم لكل شخص)\n"
+        "- /edit <تعديل> كرد على صورة لتعديلها\n"
+        "- أو رد على أي صورة بنص التعديل مباشرة بدون كتابة /edit"
+    )
+
+
+@app.on_message(filters.command("image") & in_allowed_group)
+async def image_cmd(_, message: Message):
+    prompt = message.text.split(None, 1)
+    if len(prompt) < 2 or not prompt[1].strip():
+        await message.reply_text("اكتب وصف الصورة بعد الأمر، مثال:\n/image منظر جبال عند الغروب")
+        return
+
+    user_id = message.from_user.id
+    if not check_and_increment(user_id):
+        await message.reply_text(f"وصلت للحد اليومي ({DAILY_LIMIT} صور). جرب باچر 🙏")
+        return
+
+    status = await message.reply_text("⏳ جاري توليد الصورة...")
+    img_bytes = gemini_generate(prompt[1].strip())
+
+    if img_bytes is None:
+        await status.edit_text("❌ صار خطأ أثناء توليد الصورة، حاول مرة ثانية.")
+        return
+
+    await message.reply_photo(photo=io.BytesIO(img_bytes))
+    await status.delete()
+
+
+@app.on_message(filters.command("edit") & in_allowed_group)
+async def edit_cmd(_, message: Message):
+    if not message.reply_to_message or not message.reply_to_message.photo:
+        await message.reply_text("استخدم /edit كرد (reply) على صورة مع كتابة التعديل المطلوب.")
+        return
+
+    parts = message.text.split(None, 1)
+    if len(parts) < 2 or not parts[1].strip():
+        await message.reply_text("اكتب التعديل المطلوب بعد الأمر، مثال:\n/edit خلي الخلفية زرقاء")
+        return
+
+    await _do_edit(message, message.reply_to_message, parts[1].strip())
+
+
+@app.on_message(filters.text & filters.reply & in_allowed_group & ~filters.command(["image", "edit", "start"]))
+async def implicit_edit(_, message: Message):
+    """رد نصي (بدون أمر) على صورة = طلب تعديل."""
+    replied = message.reply_to_message
+    if not replied or not replied.photo:
+        return
+    await _do_edit(message, replied, message.text.strip())
+
+
+async def _do_edit(message: Message, photo_message: Message, prompt: str):
+    user_id = message.from_user.id
+    if not check_and_increment(user_id):
+        await message.reply_text(f"وصلت للحد اليومي ({DAILY_LIMIT} صور). جرب باچر 🙏")
+        return
+
+    status = await message.reply_text("⏳ جاري تعديل الصورة...")
+
+    file_path = await photo_message.download(in_memory=True)
+    image_bytes = file_path.getvalue()
+
+    result_bytes = gemini_edit(image_bytes, prompt)
+
+    if result_bytes is None:
+        await status.edit_text("❌ صار خطأ أثناء تعديل الصورة، حاول مرة ثانية.")
+        return
+
+    await message.reply_photo(photo=io.BytesIO(result_bytes))
+    await status.delete()
+
+
+if __name__ == "__main__":
+    log.info("Bot starting...")
+    app.run()

requirements.txt

@@ -1,3 +1,3 @@
-telethon==1.36.0
-motor==3.5.1
-pymongo==4.8.0
+pyrogram==2.0.106
+tgcrypto==1.2.5
+requests==2.32.3