Update app.py

app.py

@@ -1,125 +1,126 @@
-import gradio as gr
-import selenium
-from selenium import webdriver
-from selenium.webdriver.common.keys import Keys
-from selenium.common.exceptions import NoSuchElementException
-from selenium.webdriver import Chrome, ChromeOptions
-import time
-import pandas as pd
-from selenium.webdriver.support import expected_conditions as EC
-from selenium.webdriver.common.by import By
-from selenium.webdriver.support.ui import WebDriverWait
-from time import sleep
-from webdriver_manager.chrome import ChromeDriverManager
-
-
-
-
-
-
-
-
-global name
-global password
-global topic
-
-def login_and_work(username_text, password_text, topic_text):
-    #options = webdriver.ChromeOptions()
-    #options.add_argument('--headless')
-    #options.add_argument('--no-sandbox')
-    # options.add_argument('--disable-dev-shm-usage')
-    # options.use_chromium = True
-    driver = webdriver.Chrome(ChromeDriverManager().install())
-    driver.get('https://twitter.com/i/flow/login')
-    time.sleep(5)
-    username = driver.find_element("xpath",'//input[@name="text"]')
-    username.send_keys(username_text)
-    driver.find_element("xpath",'(//*[@role="button"])[4]').click() # in prevoius block
-    
-    try:
-        time.sleep(10)
-        y = driver.find_element("xpath",'//h1//span')
-        if(y.text =='Enter your phone number or username'):
-            username2 = driver.find_element("xpath",'//input[@name="text"]')
-            username2.send_keys(username_text)
-            driver.find_element("xpath",'(//*[@role="button"])[2]').click()
-    except:
-        
-        x = 10
-
-    password =  driver.find_element("xpath",'//input[@name="password"]')
-    password.send_keys(password_text)
-    driver.find_element("xpath",'(//*[@role = "button"])[4]').click()
-
-
-    try:
-        driver.maximize_window()
-        time.sleep(5)
-        driver.maximize_window()
-        search = driver.find_element("xpath",'//input[@placeholder="Search"]')
-        search.send_keys(Keys.CONTROL + "a")
-        search.send_keys(Keys.DELETE)
-        search.send_keys(topic_text)
-        search.send_keys(Keys.RETURN)
-
-        df = pd.DataFrame()
-        df['Name']=''
-        df['Tweet'] = ''
-        #df['Sentiment'] = ''
-        tweet_count = 6
-        i = 1
-        element = driver.find_element("xpath","//body")
-        time.sleep(5)
-        while True and i <= tweet_count:
-            
-            try:
-                tweet_name =  driver.find_element("xpath",f'(//*[@data-testid="User-Name"])[{i}]')
-                df.loc[i, 'Name'] = tweet_name.text
-                    
-                tweet_div = driver.find_element("xpath",f'(//*[@data-testid="tweetText"])[{i}]')
-                if(tweet_div):
-                    #print(driver.find_element("xpath",'(//*[@data-testid="tweetText"])[1]').getText())
-                    df.loc[i, 'Tweet'] = tweet_div.text
-                    i += 1
-                element.send_keys(Keys.PAGE_DOWN)
-                time.sleep(5)
-            except: 
-                time.sleep(5)
-                tweet_name =  driver.find_element("xpath",f'(//*[@data-testid="User-Name"])[{i}]')
-                df.loc[i, 'Name'] = tweet_name.text
-                    
-                tweet_div = driver.find_element("xpath",f'(//*[@data-testid="tweetText"])[{i}]')
-                if(tweet_div):
-                    #print(driver.find_element("xpath",'(//*[@data-testid="tweetText"])[1]').getText())
-                    df.loc[i, 'Tweet'] = tweet_div.text
-                    i += 1
-                element.send_keys(Keys.PAGE_DOWN)
-                time.sleep(5)
-
-
-        #model = tweetnlp.load_model('sentiment', multilingual=True)  # Or `model = tweetnlp.Sentiment()` 
-
-        # for i in range(1,len(df)+1):
-        #     print("text is...", df.loc[i,'Tweet'])
-            
-            
-        #     #y = sentiment_pipeline(df.loc[i, 'Tweet'])
-        #     #print('label is..', y[0]['label'])
-        #     print('model sentiment is..', model.sentiment(df.loc[i, 'Tweet']))
-        #     df.loc[i, 'Sentiment'] = str(model.sentiment(df.loc[i, 'Tweet']))
-        
-        return df
-        
-    except Exception as e:
-        x = 10
-
-with gr.Blocks() as demo:
-
-    name = gr.Textbox(label="Username")
-    password = gr.Textbox(label="Password", type = 'password')
-    topic= gr.Textbox(label="Topic")
-    
-    btn = gr.Button(value="Submit")
-    btn.click(login_and_work, inputs=[name, password, topic], outputs= gr.Dataframe(headers=['Name', 'Tweet']))
-
+import gradio as gr
+import selenium
+from selenium import webdriver
+from selenium.webdriver.common.keys import Keys
+from selenium.common.exceptions import NoSuchElementException
+from selenium.webdriver import Chrome, ChromeOptions
+import time
+import pandas as pd
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from time import sleep
+from webdriver_manager.chrome import ChromeDriverManager
+
+
+
+
+
+
+
+
+global name
+global password
+global topic
+
+def login_and_work(username_text, password_text, topic_text):
+    #options = webdriver.ChromeOptions()
+    #options.add_argument('--headless')
+    #options.add_argument('--no-sandbox')
+    # options.add_argument('--disable-dev-shm-usage')
+    # options.use_chromium = True
+    options = webdriver.ChromeOptions()
+    driver = webdriver.Chrome(options = options)
+    driver.get('https://twitter.com/i/flow/login')
+    time.sleep(5)
+    username = driver.find_element("xpath",'//input[@name="text"]')
+    username.send_keys(username_text)
+    driver.find_element("xpath",'(//*[@role="button"])[4]').click() # in prevoius block
+    
+    try:
+        time.sleep(10)
+        y = driver.find_element("xpath",'//h1//span')
+        if(y.text =='Enter your phone number or username'):
+            username2 = driver.find_element("xpath",'//input[@name="text"]')
+            username2.send_keys(username_text)
+            driver.find_element("xpath",'(//*[@role="button"])[2]').click()
+    except:
+        
+        x = 10
+
+    password =  driver.find_element("xpath",'//input[@name="password"]')
+    password.send_keys(password_text)
+    driver.find_element("xpath",'(//*[@role = "button"])[4]').click()
+
+
+    try:
+        driver.maximize_window()
+        time.sleep(5)
+        driver.maximize_window()
+        search = driver.find_element("xpath",'//input[@placeholder="Search"]')
+        search.send_keys(Keys.CONTROL + "a")
+        search.send_keys(Keys.DELETE)
+        search.send_keys(topic_text)
+        search.send_keys(Keys.RETURN)
+
+        df = pd.DataFrame()
+        df['Name']=''
+        df['Tweet'] = ''
+        #df['Sentiment'] = ''
+        tweet_count = 6
+        i = 1
+        element = driver.find_element("xpath","//body")
+        time.sleep(5)
+        while True and i <= tweet_count:
+            
+            try:
+                tweet_name =  driver.find_element("xpath",f'(//*[@data-testid="User-Name"])[{i}]')
+                df.loc[i, 'Name'] = tweet_name.text
+                    
+                tweet_div = driver.find_element("xpath",f'(//*[@data-testid="tweetText"])[{i}]')
+                if(tweet_div):
+                    #print(driver.find_element("xpath",'(//*[@data-testid="tweetText"])[1]').getText())
+                    df.loc[i, 'Tweet'] = tweet_div.text
+                    i += 1
+                element.send_keys(Keys.PAGE_DOWN)
+                time.sleep(5)
+            except: 
+                time.sleep(5)
+                tweet_name =  driver.find_element("xpath",f'(//*[@data-testid="User-Name"])[{i}]')
+                df.loc[i, 'Name'] = tweet_name.text
+                    
+                tweet_div = driver.find_element("xpath",f'(//*[@data-testid="tweetText"])[{i}]')
+                if(tweet_div):
+                    #print(driver.find_element("xpath",'(//*[@data-testid="tweetText"])[1]').getText())
+                    df.loc[i, 'Tweet'] = tweet_div.text
+                    i += 1
+                element.send_keys(Keys.PAGE_DOWN)
+                time.sleep(5)
+
+
+        #model = tweetnlp.load_model('sentiment', multilingual=True)  # Or `model = tweetnlp.Sentiment()` 
+
+        # for i in range(1,len(df)+1):
+        #     print("text is...", df.loc[i,'Tweet'])
+            
+            
+        #     #y = sentiment_pipeline(df.loc[i, 'Tweet'])
+        #     #print('label is..', y[0]['label'])
+        #     print('model sentiment is..', model.sentiment(df.loc[i, 'Tweet']))
+        #     df.loc[i, 'Sentiment'] = str(model.sentiment(df.loc[i, 'Tweet']))
+        
+        return df
+        
+    except Exception as e:
+        x = 10
+
+with gr.Blocks() as demo:
+
+    name = gr.Textbox(label="Username")
+    password = gr.Textbox(label="Password", type = 'password')
+    topic= gr.Textbox(label="Topic")
+    
+    btn = gr.Button(value="Submit")
+    btn.click(login_and_work, inputs=[name, password, topic], outputs= gr.Dataframe(headers=['Name', 'Tweet']))
+
 demo.launch(share=True)