Add Target Market analysis functionality and integrate into concurrent execution

classes/client_summary.py

@@ -20,6 +20,8 @@ class CientSummary:
             st.session_state['client_name'] = ''
         if 'client_webiste' not in st.session_state:
             st.session_state['client_webiste'] = ''
+        if 'target_market' not in st.session_state:
+            st.session_state['target_market'] = ''
     
     def process (self):
             with st.spinner('Seo Analyst...', show_time=True):
@@ -38,6 +40,7 @@ class CientSummary:
 
                         if self.client_summary:
                             st.session_state['client_summary'] = 'uploaded'
+                            st.session_state['target_market'] = 'uploaded'
                             collect_telemetry(debug_client_summary)
                         if self.name:
                             st.session_state['client_name'] = 'uploaded'

classes/response_target_market.py

@@ -0,0 +1,91 @@
+from io import StringIO
+from urllib.parse import urlparse
+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
+import json
+from pymongo import MongoClient
+
+class TargetMarket:
+    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('Target Market 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' : 'Target Market Analyst', 'result': payload_txt_model}
+                                upload_response(debug_info)
+
+                                st.session_state['target_market'] = ''
+                                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_target_market = st.session_state['target_market']
+                if session_target_market == 'uploaded':
+                    count += 1
+                    self.payload += self.fetch_data("Client Summary")
+                
+            except Exception as e:
+                pass
+            
+            if count >= 1:
+                self.process()
+                                       
+
+if __name__ == "__main__":
+    st.set_page_config(layout="wide")
+
+upload = uploadFile()

pages/analyzing_page.py

@@ -12,6 +12,7 @@ from classes.response_pull_through_offers import PullThroughOffers
 from classes.response_content import Content
 from classes.response_sem_ppc import Sem_PPC
 from classes.response_marketplace import Marketplace
+from classes.response_target_market import TargetMarket
 
 def run_analysis():
     # Placeholders for status updates
@@ -25,6 +26,7 @@ def run_analysis():
     content_status = st.empty()
     sem_ppc = st.empty()
     marketplace = st.empty()
+    target_market = st.empty()
 
     def run_off_page_analysis():
         try:
@@ -126,6 +128,16 @@ def run_analysis():
             marketplace.error(f"Marketplace Analysis failed: {e}")
             return None
     
+    def run_target_market_analysis():
+        try:
+            target_market.info("Starting Target Market Analysis...")
+            result = TargetMarket(os.getenv('Model_Target_Market_Analyst'))
+            target_market.success("Target Market Analysis completed successfully.")
+            return result
+        except Exception as e:
+            target_market.error(f"Target Market Analysis failed: {e}")
+            return None
+        
     # Create threads for concurrent execution
     off_page_thread = threading.Thread(target=run_off_page_analysis)
     on_page_thread = threading.Thread(target=run_on_page_analysis)
@@ -137,6 +149,7 @@ def run_analysis():
     content_thread = threading.Thread(target=run_content)
     content_sem_ppc_thread = threading.Thread(target=run_sem_ppc_analysis)
     marketplace_thread = threading.Thread(target=run_marketplace_analysis)
+    target_market_thread = threading.Thread(target=run_target_market_analysis)
 
     # Attach Streamlit context to threads
     add_script_run_ctx(off_page_thread)
@@ -149,6 +162,7 @@ def run_analysis():
     add_script_run_ctx(content_thread)
     add_script_run_ctx(content_sem_ppc_thread)
     add_script_run_ctx(marketplace_thread)
+    add_script_run_ctx(target_market_thread)
 
     # Start threads
     off_page_thread.start()
@@ -161,6 +175,7 @@ def run_analysis():
     content_thread.start()
     content_sem_ppc_thread.start()
     marketplace_thread.start()
+    target_market_thread.start()
 
     # Wait for threads to complete
     off_page_thread.join()
@@ -173,6 +188,7 @@ def run_analysis():
     content_thread.join()
     content_sem_ppc_thread.join()
     marketplace_thread.join()
+    target_market_thread.join()
 
     st.success("🎉 All analyses completed!") # Final success message
     # --- Display Button After Completion ---