Spaces:

Opera8
/

Tsj

Sleeping

App Files Files Community

Opera8 commited on Jun 14

Commit

c996b12

verified ·

1 Parent(s): 9db1885

Update app.py

Browse files

Files changed (1) hide show

app.py +92 -33

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ app = FastAPI()
 UPLOAD_DIR = "temp_uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
-# لایه فرانت‌اند چت‌باکس با استایل شیشه‌ای (Glassmorphism)
 HTML_CHAT_INTERFACE = """
 <!DOCTYPE html>
 <html lang="fa" dir="rtl">
@@ -428,35 +428,93 @@ async def chat_endpoint(message: str = Form(...), image: UploadFile = File(None)
             await page.goto(url, wait_until="networkidle", timeout=45000)
             await page.wait_for_timeout(3000)
-            # کلیک اولیه روی کادر متنی برای اطمینان از مقداردهی و رندر کامل ساختار DOM گوگل
-            initial_input = await page.query_selector("textarea, input[type='text'], [contenteditable='true']")
-            if initial_input:
-                await initial_input.click()
-                await page.wait_for_timeout(1000)
-            # عملیات تزریق مستقیم فایل به تگ مخفی اینپوت بدون وابستگی به کلیک روی منوها
             if saved_image_path and os.path.exists(saved_image_path):
                 try:
-                    # پیدا کردن المان اصلی فایل آپلودر گوگل لنز در صفحه
-                    file_input_element = await page.query_selector("input[type='file']")
-                    if file_input_element:
-                        # آپلود مستقیم فایل روی المنت
-                        await file_input_element.set_input_files(saved_image_path)
-                        # انتظار ۵ ثانیه‌ای جهت اطمینان از اتمام آپلود و رندر شدن تامبنیل عکس در کادر گوگل
-                        await page.wait_for_timeout(5000)
-                    else:
-                        print("Warning: Google file input element not found.")
-                except Exception as img_err:
-                    print(f"Direct file injection failed: {img_err}")
-            # یافتن لایه نهایی کادر متن جهت درج دستور متنی کاربر
             input_box = None
             candidates = await page.query_selector_all("textarea, input, [contenteditable='true']")
             for el in candidates:
-                placeholder = await el.get_attribute("placeholder") or ""
-                if any(w in placeholder for w in ["بپرسید", "ask", "پیام", "چطور", "هرچه"]):
-                    input_box = el
-                    break
             if not input_box:
                 input_box = await page.query_selector("textarea") or await page.query_selector("input[type='text']")
@@ -464,28 +522,28 @@ async def chat_endpoint(message: str = Form(...), image: UploadFile = File(None)
             if input_box:
                 await input_box.focus()
                 await input_box.fill(message)
-                await page.wait_for_timeout(1000)
                 await input_box.press("Enter")
-                # بررسی کلیک پشتیبان بر روی آیکون فلش ارسال برای اطمینان از Submit فرم
                 try:
                     send_buttons = await page.query_selector_all("button, [role='button']")
                     for btn in send_buttons:
-                        aria = (await btn.get_attribute("aria-label") or "").lower()
-                        if any(w in aria for w in ["send", "ارسال", "arrow", "up"]):
-                            if await btn.is_visible():
-                                await btn.click(timeout=1000)
                                 break
                 except:
                     pass
-            # انتظار ۹ ثانیه‌ای برای پردازش نهایی تصویر و تولید متن خروجی توسط موتور هوش مصنوعی گوگل
             await page.wait_for_timeout(9000)
             full_text = await page.evaluate("() => document.body.innerText")
             await browser.close()
-            # فیلترسازی و تمیز کردن متون خروجی سرچ گوگل
             lines = full_text.split('\n')
             cleaned_lines = []
             skip_headers = ["تصاویر", "ویدیوها", "اخبار", "نقشه‌ها", "خرید کردن", "کتاب‌ها", "مالی", "حالت موضوع‌محور", "ورود"]
@@ -509,6 +567,7 @@ async def chat_endpoint(message: str = Form(...), image: UploadFile = File(None)
                 await browser.close()
             raise HTTPException(status_code=500, detail=f"خطا در رندر سرور هوش مصنوعی: {str(e)}")
         finally:
             if saved_image_path and os.path.exists(saved_image_path):
                 try:
                     os.remove(saved_image_path)

 UPLOAD_DIR = "temp_uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
+# لایه فرانت‌اند چت‌باکس فوتوریستی با استایل شیشه‌ای (Glassmorphism)
 HTML_CHAT_INTERFACE = """
 <!DOCTYPE html>
 <html lang="fa" dir="rtl">
             await page.goto(url, wait_until="networkidle", timeout=45000)
             await page.wait_for_timeout(3000)
+            # پروسه مالتی‌مدال و انتقال تصویر با ضریب ایمنی بالا و گام‌های مستقل خطا‌پذیر
             if saved_image_path and os.path.exists(saved_image_path):
+                uploaded = False
+                # استراتژی ۱: تلاش برای تزریق مستقیم به لایه بارگذاری فایل (بدون نیاز به کلیک و باز کردن منو)
                 try:
+                    file_input = await page.query_selector("input[type='file']")
+                    if file_input:
+                        await file_input.set_input_files(saved_image_path)
+                        await page.wait_for_timeout(3000)
+                        uploaded = True
+                        print("Direct file input injection succeeded.")
+                except Exception as e:
+                    print(f"Direct injection bypassed: {e}")
+                # استراتژی ۲: اگر تزریق مستقیم عمل نکرد، کلیک روی آیکون پلاس با تایم‌اوت کوتاه و باز کردن منوی گالری
+                if not uploaded:
+                    try:
+                        # فوکوس روی کادر چت با تایم‌اوت ۲ ثانیه‌ای برای باز شدن ابزارهای کناری
+                        chat_inputs = await page.query_selector_all("textarea, input[type='text'], [contenteditable='true']")
+                        for inp in chat_inputs:
+                            try:
+                                if await inp.is_visible():
+                                    await inp.click(timeout=2000)
+                                    await page.wait_for_timeout(500)
+                                    break
+                            except:
+                                continue
+                        # پیدا کردن آیکون مثبت یا دوربین و فعال‌سازی شبیه‌ساز فایل چوزر
+                        elements = await page.query_selector_all("button, div[role='button'], svg, [aria-label]")
+                        for el in elements:
+                            try:
+                                if await el.is_visible():
+                                    aria = (await el.get_attribute("aria-label") or "").lower()
+                                    html = (await el.inner_html() or "").lower()
+                                    text = (await el.inner_text() or "").lower()
+                                    combined = aria + html + text
+                                    if any(k in combined for k in ["+", "plus", "camera", "lens", "دوربین", "تصویر", "عکس"]):
+                                        async with page.expect_file_chooser(timeout=3000) as fc_info:
+                                            await el.click(timeout=2000) # تایم‌اوت امن ۲ ثانیه‌ای به جای ۳۰ ثانیه معطلی
+                                        file_chooser = await fc_info.value
+                                        await file_chooser.set_files(saved_image_path)
+                                        await page.wait_for_timeout(3000)
+                                        uploaded = True
+                                        print("Uploaded via main UI icon trigger.")
+                                        break
+                            except:
+                                continue
+                    except Exception as e:
+                        print(f"UI icon flow failed: {e}")
+                # استراتژی ۳: کلیک مستقیم روی منوی پاپ‌آپ پایینی (گالری یا فایل‌ها) در صورت باز شدن لایه شیت گوگل
+                if not uploaded:
+                    try:
+                        menu_items = await page.query_selector_all("div, span, button, p")
+                        for item in menu_items:
+                            try:
+                                item_text = (await item.inner_text() or "").strip()
+                                if item_text in ["گالری", "فایل‌ها", "Gallery", "Files"]:
+                                    if await item.is_visible():
+                                        async with page.expect_file_chooser(timeout=3000) as fc_info:
+                                            await item.click(timeout=2000)
+                                        file_chooser = await fc_info.value
+                                        await file_chooser.set_files(saved_image_path)
+                                        await page.wait_for_timeout(3000)
+                                        uploaded = True
+                                        print("Uploaded via bottom sheet item click.")
+                                        break
+                            except:
+                                continue
+                    except Exception as e:
+                        print(f"Bottom sheet workflow failed: {e}")
+            # یافتن کادر متنی گفتگو جهت وارد کردن متن پیام کاربر
             input_box = None
             candidates = await page.query_selector_all("textarea, input, [contenteditable='true']")
             for el in candidates:
+                try:
+                    if await el.is_visible():
+                        placeholder = await el.get_attribute("placeholder") or ""
+                        if any(w in placeholder for w in ["بپرسید", "ask", "پیام", "چطور", "هرچه"]):
+                            input_box = el
+                            break
+                except:
+                    continue
             if not input_box:
                 input_box = await page.query_selector("textarea") or await page.query_selector("input[type='text']")
             if input_box:
                 await input_box.focus()
                 await input_box.fill(message)
+                await page.wait_for_timeout(500)
                 await input_box.press("Enter")
+                # کلیک بر روی آیکون ارسال فرم (فلش آبی بالا) با استفاده از زمان‌بندی محدود ۲ ثانیه‌ای
                 try:
                     send_buttons = await page.query_selector_all("button, [role='button']")
                     for btn in send_buttons:
+                        if await btn.is_visible():
+                            aria = (await btn.get_attribute("aria-label") or "").lower()
+                            if any(w in aria for w in ["send", "ارسال", "arrow", "up"]):
+                                await btn.click(timeout=2000)
                                 break
                 except:
                     pass
+            # انتظار تعمدی به مدت ۹ ثانیه جهت ت��یپ کامل جواب توسط هوش مصنوعی گوگل
             await page.wait_for_timeout(9000)
             full_text = await page.evaluate("() => document.body.innerText")
             await browser.close()
+            # فیلترسازی خطوط خروجی برای تمیز کردن تگ‌های اضافی سرچ گوگل
             lines = full_text.split('\n')
             cleaned_lines = []
             skip_headers = ["تصاویر", "ویدیوها", "اخبار", "نقشه‌ها", "خرید کردن", "کتاب‌ها", "مالی", "حالت موضوع‌محور", "ورود"]
                 await browser.close()
             raise HTTPException(status_code=500, detail=f"خطا در رندر سرور هوش مصنوعی: {str(e)}")
         finally:
+            # پاکسازی حتمی فایل تصویر از روی سرور داکر
             if saved_image_path and os.path.exists(saved_image_path):
                 try:
                     os.remove(saved_image_path)