Spaces:
Build error
Build error
File size: 4,832 Bytes
e23525d 76e7594 e23525d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | 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
from helper.data_field import get_analyst_response
class ExecutiveSummary:
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('Executive Summary...', 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' : 'Executive Summary', 'result': payload_txt_model}
upload_response(debug_info)
except Exception as e:
pass
st.session_state['analyzing'] = False
def row1(self):
st.session_state['analyzing'] = False
self.payload = ""
self.website_and_tools_data = get_analyst_response("Website and Tools Analyst")
self.sem_data = get_analyst_response("SEM/PPC Analyst")
self.seo_data = get_analyst_response("SEO Analyst")
self.on_page_data = get_analyst_response("On Page Analyst")
self.off_page_data = get_analyst_response("Off Page Analyst")
self.social_media_data = get_analyst_response("Social Media Analyst")
self.content_data = get_analyst_response("Content Analyst")
self.marketpalce_data = get_analyst_response("Marketplace Analyst")
self.target_market_data = get_analyst_response("Target Market Analyst")
self.website_audience_data = get_analyst_response("Pull through offers Analyst")
self.pull_through_data = get_analyst_response("Website Audience Acquisition Analyst")
self.lld_data = get_analyst_response("LLD/PM/LN Analyst")
self.pna_data = get_analyst_response("Content - Process and Assets Analyst")
analyst_data_dict = {
"Website and Tools": self.website_and_tools_data,
"SEM/PPC": self.sem_data,
"SEO": self.seo_data,
"On Page": self.on_page_data,
"Off Page": self.off_page_data,
"Social Media": self.social_media_data,
"Content": self.content_data,
"Marketplace": self.marketpalce_data,
"Target Market": self.target_market_data,
"Pull through offers": self.website_audience_data,
"Website Audience Acquisition": self.pull_through_data,
"LLD/PM/LN": self.lld_data,
"Content - Process and Assets": self.pna_data
}
for analyst_name, data in analyst_data_dict.items():
self.payload += f"\n\n--- {analyst_name} Analysis ---\n"
if isinstance(data, list):
self.payload += "\n".join(map(str, data))
else:
self.payload += str(data)
self.process()
if __name__ == "__main__":
st.set_page_config(layout="wide")
upload = uploadFile()
|