Update app.py

app.py

@@ -3,14 +3,21 @@ import google.generativeai as genai
 import ast
 import time
 import os
-# import re
-# import json
+import re
 
 st.title('Verified Interseting fact geneartor')
 
 secret_key = os.getenv("SECRET_KEY")
 genai.configure(api_key=secret_key)
 
+def extract_python_code(text):
+    pattern = r"```python\n(.*?)```"
+    match = re.search(pattern, text, re.DOTALL)
+    if match:
+        return match.group(1).strip()
+    else:
+        return None
+
 
 def get_gemini_response(input):
     prompt2='''you are an fact checker,you will get an text.
@@ -43,44 +50,12 @@ text - '''
 def check(x):
     try:
         y=ast.literal_eval(x)
-        print(y)
         return y
     except:
         y=''
         st.warning('rerun code not splitted correctly')
         return y
 
-# if text:
-#     new=prompt1+text
-#     model=genai.GenerativeModel('gemini-1.5-flash')
-#     response=model.generate_content(new)
-
-#     a=response.text
-#     print(a)
-#     b=a[10:-5]
-#     print(b)
-
-
-
-#     # try:
-#     #     my_list = ast.literal_eval(b)
-#     #     print(my_list)
-#     # except:
-#     #     my_list=''
-#     #     st.warning('rerun code not splitted correctly')
-
-#     my_list=check(b)
-
-#     if isinstance(my_list,list):
-#         for i in my_list:
-#             c=get_gemini_response(i)
-#             new_list=check(c)
-#             if new_list[0]=='True':
-#                 st.write(i)
-#                 st.write(c)
-#             time.sleep(2)
-#     else:
-#         st.warning('rerun code error in checking')
 
 
 submit=st.button('submit')
@@ -91,7 +66,7 @@ if text and submit:
         model = genai.GenerativeModel('gemini-1.5-flash')
         response = model.generate_content(new)
         a = response.text
-        b = a[10:-5]
+        b = extract_python_code(a)
         my_list = check(b)
         if isinstance(my_list, list):
             for i in my_list: