2 files changed, 42 insertions, 20 deletions

diff --git a/gui/streamlit_app.py b/gui/streamlit_app.py
index ff1cb6d4..41b75f4b 100644
--- a/gui/streamlit_app.py
+++ b/gui/streamlit_app.py
@@ -1,25 +1,29 @@
 import streamlit as st
 import phind
 
-phind.cf_clearance = ''
-phind.user_agent   = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
+# Set cloudflare clearance and user agent
+phind.cloudflare_clearance = ''
+phind.phind_api = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
 
-def phind_get_answer(question:str)->str:
-    # set cf_clearance cookie
+
+def get_answer(question: str) -> str:
+    # Set cloudflare clearance cookie and get answer from GPT-4 model
     try:
-    
         result = phind.Completion.create(
-        model  = 'gpt-4',
-        prompt = question,
-        results     = phind.Search.create(question, actualSearch = True),
-        creative    = False,
-        detailed    = False,
-        codeContext = '') 
+            model='gpt-4',
+            prompt=question,
+            results=phind.Search.create(question, actualSearch=True),
+            creative=False,
+            detailed=False,
+            codeContext=''
+        )
         return result.completion.choices[0].text
-
     except Exception as e:
-        return 'An error occured, please make sure you are using a cf_clearance token and correct useragent | %s' % e
+        # Return error message if an exception occurs
+        return f'An error occurred: {e}. Please make sure you are using a valid cloudflare clearance token and user agent.'
+
 
+# Set page configuration and add header
 st.set_page_config(
     page_title="gpt4freeGUI",
     initial_sidebar_state="expanded",
@@ -30,19 +34,21 @@ st.set_page_config(
         'About': "### gptfree GUI"
     }
 )
-
 st.header('GPT4free GUI')
 
-question_text_area = st.text_area('🤖 Ask Any Question :', placeholder='Explain quantum computing in 50 words')
+# Add text area for user input and button to get answer
+question_text_area = st.text_area(
+    '🤖 Ask Any Question :', placeholder='Explain quantum computing in 50 words')
 if st.button('🧠 Think'):
-    answer = phind_get_answer(question_text_area)
+    answer = get_answer(question_text_area)
+    # Display answer
     st.caption("Answer :")
     st.markdown(answer)
 
-
+# Hide Streamlit footer
 hide_streamlit_style = """
             <style>
             footer {visibility: hidden;}
             </style>
             """
-st.markdown(hide_streamlit_style, unsafe_allow_html=True) 
\ No newline at end of file
+st.markdown(hide_streamlit_style, unsafe_allow_html=True)
diff --git a/quora/mail.py b/quora/mail.py
index c4707860..e6ce96c2 100644
--- a/quora/mail.py
+++ b/quora/mail.py
@@ -38,7 +38,7 @@ class Emailnator:
         return self.email
 
     def get_message(self):
-        print("waiting for code...")
+        print("Waiting for message...")
 
         while True:
             sleep(2)
@@ -49,6 +49,7 @@ class Emailnator:
             mail_token = loads(mail_token.text)["messageData"]
 
             if len(mail_token) == 2:
+                print("Message received!")
                 print(mail_token[1]["messageID"])
                 break
 
@@ -63,4 +64,19 @@ class Emailnator:
         return mail_context.text
 
     def get_verification_code(self):
-        return findall(r';">(\d{6,7})</div>', self.get_message())[0]
+        message = self.get_message()
+        code = findall(r';">(\d{6,7})</div>', message)[0]
+        print(f"Verification code: {code}")
+        return code
+
+    def clear_inbox(self):
+        print("Clearing inbox...")
+        self.client.post(
+            "https://www.emailnator.com/delete-all",
+            json={"email": self.email},
+        )
+        print("Inbox cleared!")
+
+    def __del__(self):
+        if self.email:
+            self.clear_inbox()