cleaning up

pag/monitor.py

@@ -17,7 +17,13 @@ import grpc
 import pb.timesfm_pb2_grpc
 import pb.timesfm_pb2
 from openai import OpenAI
+from dotenv import load_dotenv
 import time
+import dotenv
+
+
+load_dotenv()
+
 
 def check_authentication():
     if not check_password():
@@ -30,8 +36,8 @@ config.instance_id       = '352670fb-2edf-4abd-90c8-437485a2403e'
 config.sh_client_id      = 'ca95f10f-443c-4c60-9a36-98950292bb9b'
 config.sh_client_secret  = 'rNFGRxGNiNFrXJfGyHIkVRyGOrdWNsfI'
 config.sh_timesfm_IP = "34.121.141.161"
-client = OpenAI(api_key= os.environ['OPENAI_KEY'])
-print(os.environ)
+OpenAI_key = os.getenv('OPENAI_KEY')
+client = OpenAI(api_key= OpenAI_key)
 
 def select_field(gdf):
     st.markdown("""
@@ -309,6 +315,7 @@ def monitor_fields():
             st.write(f"Press the button below to predict {metric} for the next 10 weeks")
             subcol1, subcol2, subcol3 = st.columns(3)
             if subcol2.button(f'Predict & Recommend'):
+                
                 start_date = '2024-01-01'
                 today = datetime.today()
                 end_date = today.strftime('%Y-%m-%d')
@@ -328,6 +335,7 @@ def monitor_fields():
                     counter = counter + 100/(len(dates))
                     my_bar.progress(round(counter), text=f"Downloading Data for the last {len(dates)//4} months: {round(counter)}%")
 
+                st.subheader('Predictions:')
                 # chart_data = pd.DataFrame(
                 # {
                 #     "date": [metric[0] for metric in downloaded_prev_metrics],
@@ -379,6 +387,7 @@ def monitor_fields():
 
                 st.area_chart(prediction_chart_data, x="date", y=[f"history_{metric}_values", f"predicted_{metric}_values"])
 
+                st.subheader('Recommendation:')
                 crop = "Wheat"
 
                 try:
@@ -397,9 +406,9 @@ def monitor_fields():
                         frequency_penalty=0,
                         presence_penalty=0
                     )
-                    st.write(response.choices[0].message.content)
+                    st.caption(response.choices[0].message.content)
                 except:
-                    st.write("Server Error: Could't generate recommendation!")
+                    st.code("Server Error: Could't generate recommendation!")
 
 
         

requirements.txt

@@ -12,4 +12,5 @@ streamlit==1.29.0
 geopy==2.4.1
 grpcio
 grpcio-tools
-openai
+openai
+python-dotenv