Spaces:

Opera8
/

Tsj

Sleeping

App Files Files Community

Opera8 commited on Jun 14

Commit

30181ae

verified ·

1 Parent(s): c996b12

Update app.py

Browse files

Files changed (1) hide show

app.py +64 -87

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">
@@ -425,94 +425,56 @@ async def chat_endpoint(message: str = Form(...), image: UploadFile = File(None)
             )
             page = await context.new_page()
-            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
@@ -522,40 +484,56 @@ 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(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 = ["تصاویر", "ویدیوها", "اخبار", "نقشه‌ها", "خرید کردن", "کتاب‌ها", "مالی", "حالت موضوع‌محور", "ورود"]
             for line in lines:
                 line_str = line.strip()
                 if not line_str:
                     continue
-                if any(header in line_str for header in skip_headers):
                     continue
-                if "در پاسخ‌های" in line_str or "بیشتر بدانید" in line_str:
                     continue
                 cleaned_lines.append(line_str)
             ai_response = "\n".join(cleaned_lines)
@@ -567,7 +545,6 @@ 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">
             )
             page = await context.new_page()
+            # لود اولیه سبک domcontentloaded به جای networkidle جهت افزایش سرعت
+            await page.goto(url, wait_until="domcontentloaded", timeout=30000)
+            # پیدا کردن کادر متن فرم برای آمادگی تزریق
+            try:
+                await page.wait_for_selector("textarea, input[type='text']", timeout=4000)
+            except:
+                pass
+            # پروسه مالتی‌مدال خطا‌پذیر بدون قفل شدن مرورگر
             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(1500) # زمان کوتاه فقط برای متصل شدن فایل به فرم گوگل
                         uploaded = True
                 except Exception as e:
+                    print(f"Direct injection warning: {e}")
                 if not uploaded:
                     try:
+                        elements = await page.query_selector_all("button, div[role='button'], svg")
                         for el in elements:
                             try:
+                                aria = (await el.get_attribute("aria-label") or "").lower()
+                                html = (await el.inner_html() or "").lower()
+                                combined = aria + html
+                                if any(k in combined for k in ["+", "plus", "camera", "lens", "دوربین", "تصویر"]):
+                                    async with page.expect_file_chooser(timeout=2000) as fc_info:
+                                        await el.click(timeout=1500)
+                                    file_chooser = await fc_info.value
+                                    await file_chooser.set_files(saved_image_path)
+                                    await page.wait_for_timeout(1500)
+                                    break
                             except:
                                 continue
                     except Exception as e:
+                        print(f"UI upload sequence warning: {e}")
+            # درج و ارسال نهایی پیام کاربر
             input_box = None
             candidates = await page.query_selector_all("textarea, input, [contenteditable='true']")
             for el in candidates:
                 try:
+                    placeholder = await el.get_attribute("placeholder") or ""
+                    if any(w in placeholder for w in ["بپرسید", "ask", "پیام", "چطور", "هرچه"]):
+                        input_box = el
+                        break
                 except:
                     continue
             if input_box:
                 await input_box.focus()
                 await input_box.fill(message)
+                await page.wait_for_timeout(300)
                 await input_box.press("Enter")
+            # 🚀 الگوریتم پایش داینامیک قد کشیدن متن (حذف ۹ ثانیه خواب اجباری)
+            await page.wait_for_timeout(2500) # مهلت اولیه برای استارت خوردن استریم جواب توسط گوگل
+            last_len = 0
+            stable_count = 0
+            for _ in range(20): # حداکثر ۱۰ ثانیه تلاش مجدد در گام‌های ۵۰۰ میلی‌ثانیه‌ای
+                current_text = await page.evaluate("() => document.body.innerText")
+                current_len = len(current_text or "")
+                if current_len == last_len and current_len > 0:
+                    stable_count += 1
+                    if stable_count >= 2: # اگر طول متن در ۲ دوره متوالی ثابت ماند، استریم به پایان رسیده است
+                        break
+                else:
+                    stable_count = 0
+                    last_len = current_len
+                await page.wait_for_timeout(500)
             full_text = await page.evaluate("() => document.body.innerText")
             await browser.close()
+            # 🎯 الگوریتم پاکسازی کدهای اضافی و جداسازی پاسخ خالص هوش مصنوعی
             lines = full_text.split('\n')
             cleaned_lines = []
+            exclude_keywords = [
+                "حالت هوشواره‌ای", "ویدئوها", "نقشه‌ها", "اخبار", "تصاویر", "خرید کردن", "کتاب‌ها", "مالی",
+                "All items removed from input context", "ارسال", "هرچه می‌خواهید بپرسید", "بیشتر بدانید",
+                "اگر امکان دارد آن را به فارسی ایمیل کنید", "پاسخ «حالت هوشواره‌ای» آماده است", "همه"
+            ]
             for line in lines:
                 line_str = line.strip()
                 if not line_str:
                     continue
+                # حذف خط اکو شده‌ی خود کلمه سرچ شده کاربر در ابتدای بدنه
+                if line_str == message.strip():
                     continue
+                # نقطه قطع سابمیت: رسیدن به دکمه‌های لایک/دیس‌لایک یا کادر کپی‌رایت انتهای چت گوگل
+                if any(stop_word in line_str for stop_word in ["در پاسخ‌های هوشواره", "هوشواره ممکن است اشتباه", "پاسخ‌های مربوط به افراد"]):
+                    break
+                if any(ex in line_str for ex in exclude_keywords):
                     continue
                 cleaned_lines.append(line_str)
             ai_response = "\n".join(cleaned_lines)
                 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)