Add Content_Process_and_Assets_Analyst class and integrate into analyzing_page

classes/response_content_process_and_assets_analyst.py

@@ -0,0 +1,100 @@
+import streamlit as st
+import requests
+from dotenv import load_dotenv
+import os
+from helper.upload_response import upload_response
+from helper.upload_File import uploadFile
+from pymongo import MongoClient
+
+class Content_Process_and_Assets_Analyst:
+    def __init__(self, model_url):
+        self.uploaded_files = []
+        self.file_dict = {}
+        self.model_url = model_url
+        #self.analyst_name = analyst_name
+        #self.data_src = data_src
+        #self.analyst_description = analyst_description
+        self.initialize()
+        self.row1()
+
+    def initialize(self):
+        # FOR ENV
+        load_dotenv()
+
+        # AGENT NAME
+        #st.header(self.analyst_name)
+    
+    def request_model(self, payload_txt, headers):
+        response = requests.post(self.model_url, json=payload_txt, headers=headers)
+        response.raise_for_status()
+        output = response.json()
+        st.write(output)
+        text = output["outputs"][0]["outputs"][0]["results"]["text"]["data"]["text"]
+        #text = json.loads(text)
+        st.write(text)
+        return text
+    
+    def fetch_data(self, data_field):
+        mongodb_uri = os.getenv("MONGODB_URI")
+        myclient = MongoClient(mongodb_uri)
+        mydb = myclient.get_database()
+        mycol = mydb["df_data"]
+        
+        # Sort by timestamp field in descending order
+        x = mycol.find_one(
+            {"data_field": data_field},
+            sort=[("timestamp", -1)]  
+        )
+        
+        x = x["result"]
+        return x
+    
+    def process(self):
+                with st.spinner('Content - Process and Assets Analyst...', show_time=True):
+                        st.write('')
+                        headers = {"Content-Type": "application/json", "x-api-key": f"{os.getenv('x-api-key')}"}         
+                        try:
+                                payload_txt = {"input_value": self.payload, "output_type": "text", "input_type": "chat"}
+                                payload_txt_model = self.request_model(payload_txt, headers)
+                                debug_info = {'data_field' : 'Content - Process and Assets Analyst', 'result': payload_txt_model}
+                                upload_response(debug_info)
+
+                                count = 0
+                        except Exception as e:
+                            pass
+                        st.session_state['analyzing'] = False    
+                      
+    def row1(self):
+            st.session_state['analyzing'] = False
+            self.payload = ""  
+            count = 0
+            try:
+                session_client_summary = st.session_state['client_summary']
+                if session_client_summary == 'uploaded':
+                    count += 1
+                    self.payload += self.fetch_data("Client Summary")
+            except Exception as e:
+                pass
+            try:
+                session_content_in_the_website = st.session_state['content_in_the_website']
+                if session_content_in_the_website == 'uploaded':
+                    count += 1
+                    self.payload += self.fetch_data("Content in the Website")
+            except Exception as e:
+                pass
+            try:
+                session_content_outside_the_website = st.session_state['content_outside_the_website']
+                if session_content_outside_the_website == 'uploaded':
+                    count += 1
+                    self.payload += self.fetch_data("Content outside the Website")
+            except Exception as e:
+                pass
+            
+            if count >= 3:
+                self.process()
+                                       
+
+if __name__ == "__main__":
+    st.set_page_config(layout="wide")
+
+upload = uploadFile()

pages/analyzing_page.py

@@ -18,6 +18,7 @@ from classes.response_df_overview import dfOverview
 from classes.response_desired_outcome import DesiredOutcome
 from classes.response_conversion_analyst import ConversionAnalyst
 from classes.response_website_audience_acquisition import WebsiteAudienceAcquisition
+from classes.response_content_process_and_assets_analyst import Content_Process_and_Assets_Analyst
 from classes.response_executive_summary import ExecutiveSummary
 from classes.response_snapshot import Snapshot
 
@@ -74,6 +75,7 @@ def run_analysis():
         "desired_outcome": st.empty(),
         "conversion": st.empty(),
         "website_audience": st.empty(),
+        "content_process_and_assets": st.empty(),
         "snapshot": st.empty(),
         "executive_summary": st.empty(),
         
@@ -249,6 +251,17 @@ def run_analysis():
             handler.update_error(f"Website Audience Acquisition Analysis failed: {str(e)}")
             return None
         
+    def run_content_process_and_assets_analysis():
+        handler = handlers["content_process_and_assets"]
+        try:
+            handler.update_info("Running Content - Process and Assets Analysis...")
+            result = Content_Process_and_Assets_Analyst(os.getenv('Model_Content_Process_and_Assets_Analyst'))
+            handler.update_success("Content - Process and Assets Analysis completed successfully.")
+            return result
+        except Exception as e:
+            handler.update_error(f"Content - Process and Assets Analysis failed: {str(e)}")
+            return None
+
     def run_snapshot_analysis():
         handler = handlers["snapshot"]
         try:
@@ -286,6 +299,7 @@ def run_analysis():
         (run_target_market_analysis, "target_market"),
         (run_df_overview_analysis, "df_overview"),
         (run_desired_outcomes_analysis, "desired_outcome"),
+        (run_content_process_and_assets_analysis, "content_process_and_assets"),
         (run_conversion_analysis, "conversion"),
         (run_website_audience, "website_audience")
     ]