modify app.py

app.py

@@ -5,7 +5,6 @@ from yahoo_fin import options
 import pandas_ta as ta
 from datetime import datetime, timedelta
 import requests
-import json
 from transformers import pipeline
 
 
@@ -212,35 +211,23 @@ if st.button("Analyze"):
 
         with st.spinner('Predicting...'):
             prompt_generated = prompt_generator(selected_company, option_stats, technical_indicators)
+            start_time = datetime.now()
             response = send_prompt_to_api(prompt_generated)
+            end_time = datetime.now()
 
             if response["success"]:
                 try:
+                    st.write("Raw response")
+                    st.write(response)
+                    st.write("----------------------------------------------------")
+                    st.write("Below is the generate text")
                     generated_text = response["text"]
                     st.write("Our evaluation:")
                     st.write(generated_text)
                     st.markdown("### Disclaimer:")
                     st.write(disclaimer)
-                # else:
-                #    st.error(f"Error: {response['error_message']}")
-                    # inner_json_str = json.loads(response.text)
-                    # api_response = json.loads(inner_json_str)
-
-                    # if api_response:
-                    #     for item in api_response:
-                    #         if 'generated_text' in item:
-                    #             generated_text = item['generated_text']
-                    #             st.write("API Response:")
-                    #             st.write(generated_text)
-                    #             st.markdown("### Disclaimer:")
-                    #             st.write(disclaimer)
-                    #             break
-
-                    #     else:
-                    #         st.error("Error: 'generated_text' not found in any item of the API response.")
-
-                    # else:
-                    #     st.error("Error: Empty API response.")
+                    training_duration = end_time - start_time
+                    st.write(f"Prompt generation started at : {start_time} and Prompt generation ended at : {end_time} and Total prompt generation duration : {training_duration}")
 
                 except Exception as e:
                     st.error(f"Error: Failed due to {e}")