Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import random
 from transformers import pipeline
 import pandas as pd
+from datetime import datetime
+import pytz
 
 generator = pipeline('text-generation', model='gpt2')
 max_length = 50
@@ -46,9 +48,18 @@ if st.button('Generate Prompt and Solution'):
         prompt_text, prompt, solution = generate_prompt(option)
         results.append([prompt_text, prompt, solution])
         
+    user_timezone = st.text_input("Enter your timezone (e.g., 'America/New_York'):")
+    
+    try:
+        tz = pytz.timezone(user_timezone)
+        current_time = datetime.now(tz).strftime('%Y-%m-%d %H:%M:%S %Z')
+    except Exception:
+        current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+        st.warning('Invalid timezone entered. Using the server timezone.')
+    
     with open('results.txt', 'a') as f:
         for result in results:
-            f.write(f"{result[0]}\t{result[1]}\t{result[2]}\n")
+            f.write(f"{current_time}\t{result[0]}\t{result[1]}\t{result[2]}\n")
     
-    df = pd.read_csv('results.txt', sep='\t', header=None, names=['Input', 'Prompt', 'Solution'])
+    df = pd.read_csv('results.txt', sep='\t', header=None, names=['Timestamp', 'Input', 'Prompt', 'Solution'])
     st.write(df)