Spaces:
Runtime error
Runtime error
Upload folder using huggingface_hub
Browse files
ui/app.py
CHANGED
|
@@ -3,8 +3,10 @@ import os
|
|
| 3 |
import glob
|
| 4 |
import asyncio
|
| 5 |
import sys
|
|
|
|
| 6 |
|
| 7 |
# Add project root to sys.path
|
|
|
|
| 8 |
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
| 9 |
|
| 10 |
from appagents.research_agent import MarketResearchAgent
|
|
@@ -24,7 +26,13 @@ def load_prompts(folder="prompts"):
|
|
| 24 |
prompt_lables.append(os.path.basename(file_path).replace("_", " ").replace(".txt", "").title())
|
| 25 |
return prompts, prompt_lables
|
| 26 |
|
| 27 |
-
prompts
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
# -----------------------------
|
| 30 |
# Streamlit page config
|
|
@@ -32,67 +40,166 @@ prompts, prompt_labels = load_prompts()
|
|
| 32 |
st.set_page_config(page_title="AI Chat", layout="wide")
|
| 33 |
|
| 34 |
# -----------------------------
|
| 35 |
-
# Custom CSS (
|
| 36 |
# -----------------------------
|
| 37 |
st.markdown("""
|
| 38 |
<style>
|
| 39 |
-
.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
header[data-testid="stHeader"] { display: none !important; }
|
| 41 |
|
| 42 |
-
/*
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
border-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
}
|
| 57 |
-
.hero-logo { font-size: 3.2rem; margin-right: 18px; animation: pulse 2s infinite; }
|
| 58 |
-
@keyframes pulse { 0% { transform: scale(1); opacity: 1; } 50% { transform: scale(1.1); opacity: 0.9; } 100% { transform: scale(1); opacity: 1; } }
|
| 59 |
-
.hero-text { font-size: 1.8rem; font-weight: 700; letter-spacing: 0.6px; }
|
| 60 |
-
.hero-subtext { font-size: 1rem; opacity: 0.9; margin-top: 0.25rem; }
|
| 61 |
|
| 62 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
-
/* Chat
|
| 65 |
.chat-container {
|
| 66 |
display: flex;
|
| 67 |
flex-direction: column;
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
.user-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
padding: 10px
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
| 94 |
-
|
| 95 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 96 |
}
|
| 97 |
</style>
|
| 98 |
""", unsafe_allow_html=True)
|
|
@@ -110,80 +217,155 @@ if "send_triggered" not in st.session_state:
|
|
| 110 |
st.session_state.send_triggered = False
|
| 111 |
|
| 112 |
# -----------------------------
|
| 113 |
-
# Function to fetch AI response
|
| 114 |
# -----------------------------
|
| 115 |
async def get_ai_response(prompt):
|
| 116 |
-
agent
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 120 |
|
| 121 |
# -----------------------------
|
| 122 |
# Sidebar with prompts
|
| 123 |
# -----------------------------
|
| 124 |
st.sidebar.title("💡 Quick Prompts")
|
|
|
|
| 125 |
for idx, prompt_text in enumerate(prompts):
|
| 126 |
-
|
| 127 |
if st.sidebar.button(prompt_labels[idx], key=f"prompt_{idx}", help=prompt_text):
|
| 128 |
st.session_state.input_value = prompt_text
|
| 129 |
st.session_state.send_triggered = True
|
| 130 |
|
| 131 |
# -----------------------------
|
| 132 |
-
#
|
| 133 |
# -----------------------------
|
| 134 |
st.markdown("""
|
| 135 |
-
<div class="
|
| 136 |
-
<
|
| 137 |
-
<div>
|
| 138 |
-
<div class="hero-text">AI Chatbot</div>
|
| 139 |
-
<div class="hero-subtext">Your intelligent assistant for insights, trends, and strategy exploration.</div>
|
| 140 |
-
</div>
|
| 141 |
</div>
|
| 142 |
""", unsafe_allow_html=True)
|
| 143 |
|
| 144 |
# -----------------------------
|
| 145 |
-
# Chat input
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 146 |
# -----------------------------
|
|
|
|
| 147 |
with st.form(key="chat_form", clear_on_submit=True):
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 155 |
|
| 156 |
# -----------------------------
|
| 157 |
# Handle sending message
|
| 158 |
# -----------------------------
|
| 159 |
-
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 165 |
st.session_state.input_value = ""
|
| 166 |
st.session_state.send_triggered = False
|
| 167 |
-
|
| 168 |
-
# -----------------------------
|
| 169 |
-
# Chat history display
|
| 170 |
-
# -----------------------------
|
| 171 |
-
if st.session_state.chat_history:
|
| 172 |
-
chat_html = '<div class="chat-container">'
|
| 173 |
-
for chat in st.session_state.chat_history:
|
| 174 |
-
if chat["role"] == "user":
|
| 175 |
-
chat_html += (
|
| 176 |
-
f"<div style='display:flex; justify-content:flex-end; align-items:center;'>"
|
| 177 |
-
f"<div class='user-bubble'>{chat['message']}</div>"
|
| 178 |
-
f"<span style='font-size:34px; margin-left:8px;'>👤</span>"
|
| 179 |
-
f"</div>"
|
| 180 |
-
)
|
| 181 |
-
else:
|
| 182 |
-
chat_html += (
|
| 183 |
-
f"<div style='display:flex; justify-content:flex-start; align-items:center;'>"
|
| 184 |
-
f"<span style='font-size:34px; margin-right:8px;'>🤖</span>"
|
| 185 |
-
f"<div class='ai-bubble'>{chat['message']}</div>"
|
| 186 |
-
f"</div>"
|
| 187 |
-
)
|
| 188 |
-
chat_html += '</div>'
|
| 189 |
-
st.markdown(chat_html, unsafe_allow_html=True)
|
|
|
|
| 3 |
import glob
|
| 4 |
import asyncio
|
| 5 |
import sys
|
| 6 |
+
import textwrap
|
| 7 |
|
| 8 |
# Add project root to sys.path
|
| 9 |
+
# Retaining original path logic for environment compatibility
|
| 10 |
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
| 11 |
|
| 12 |
from appagents.research_agent import MarketResearchAgent
|
|
|
|
| 26 |
prompt_lables.append(os.path.basename(file_path).replace("_", " ").replace(".txt", "").title())
|
| 27 |
return prompts, prompt_lables
|
| 28 |
|
| 29 |
+
# Assuming 'prompts' folder exists relative to where this script is run
|
| 30 |
+
try:
|
| 31 |
+
prompts, prompt_labels = load_prompts()
|
| 32 |
+
except:
|
| 33 |
+
# Fallback if prompts folder is not available
|
| 34 |
+
prompts = ["What are the top 3 market trends in AI?", "Analyze the competitive landscape for EV startups.", "Suggest a strategic pivot for a struggling e-commerce company."]
|
| 35 |
+
prompt_labels = ["AI Trends Analysis", "EV Competitive Map", "E-commerce Strategy"]
|
| 36 |
|
| 37 |
# -----------------------------
|
| 38 |
# Streamlit page config
|
|
|
|
| 40 |
st.set_page_config(page_title="AI Chat", layout="wide")
|
| 41 |
|
| 42 |
# -----------------------------
|
| 43 |
+
# Custom CSS (Perplexity-inspired UI)
|
| 44 |
# -----------------------------
|
| 45 |
st.markdown("""
|
| 46 |
<style>
|
| 47 |
+
/* 1. Global & Layout Reset */
|
| 48 |
+
.stApp { background-color: #f7f9fc; font-family: 'Inter', sans-serif; }
|
| 49 |
+
.block-container {
|
| 50 |
+
padding-top: 1.5rem !important;
|
| 51 |
+
padding-bottom: 5rem;
|
| 52 |
+
max-width: 1000px; /* Constrain content width for focus */
|
| 53 |
+
margin: 0 auto;
|
| 54 |
+
}
|
| 55 |
header[data-testid="stHeader"] { display: none !important; }
|
| 56 |
|
| 57 |
+
/* 2. Accent Color & Typography */
|
| 58 |
+
:root {
|
| 59 |
+
--accent-color: #4a5cff; /* Modern Perplexity Blue/Purple */
|
| 60 |
+
--text-color: #1c2732;
|
| 61 |
+
--secondary-text: #6b7787;
|
| 62 |
+
--card-bg: #ffffff;
|
| 63 |
+
--border-color: #e6e9ee;
|
| 64 |
+
}
|
| 65 |
+
|
| 66 |
+
/* 3. Sidebar (Quick Prompts Navigation) */
|
| 67 |
+
section[data-testid="stSidebar"] {
|
| 68 |
+
background-color: var(--card-bg);
|
| 69 |
+
border-right: 1px solid var(--border-color);
|
| 70 |
+
padding-top: 2rem !important;
|
| 71 |
+
}
|
| 72 |
+
[data-testid="stSidebar"] .stButton button {
|
| 73 |
+
text-align: left;
|
| 74 |
+
color: var(--secondary-text);
|
| 75 |
+
background-color: transparent;
|
| 76 |
+
border: none;
|
| 77 |
+
box-shadow: none;
|
| 78 |
+
padding: 10px 15px;
|
| 79 |
+
margin-bottom: 5px;
|
| 80 |
+
transition: background-color 0.2s;
|
| 81 |
+
}
|
| 82 |
+
[data-testid="stSidebar"] .stButton button:hover {
|
| 83 |
+
background-color: #eff3f8;
|
| 84 |
+
color: var(--text-color);
|
| 85 |
+
}
|
| 86 |
+
[data-testid="stSidebar"] .stButton button:active {
|
| 87 |
+
background-color: #e0e6ed;
|
| 88 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
|
| 90 |
+
/* 4. Main Header/Title */
|
| 91 |
+
.app-title-container {
|
| 92 |
+
text-align: center;
|
| 93 |
+
padding: 3rem 0 2rem 0;
|
| 94 |
+
}
|
| 95 |
+
.app-title {
|
| 96 |
+
font-size: 2.5rem;
|
| 97 |
+
font-weight: 800;
|
| 98 |
+
color: var(--text-color);
|
| 99 |
+
letter-spacing: -0.5px;
|
| 100 |
+
}
|
| 101 |
+
.app-subtitle {
|
| 102 |
+
font-size: 1.1rem;
|
| 103 |
+
color: var(--secondary-text);
|
| 104 |
+
margin-top: 0.5rem;
|
| 105 |
+
}
|
| 106 |
+
|
| 107 |
+
/* 5. Input Form (The Perplexity Search Bar) */
|
| 108 |
+
[data-testid="stForm"] {
|
| 109 |
+
padding: 1.5rem 0;
|
| 110 |
+
margin-bottom: 20px;
|
| 111 |
+
position: sticky;
|
| 112 |
+
top: 0;
|
| 113 |
+
background-color: #f7f9fc;
|
| 114 |
+
z-index: 10;
|
| 115 |
+
border-radius: 12px;
|
| 116 |
+
}
|
| 117 |
+
[data-testid="stForm"] input {
|
| 118 |
+
border-radius: 12px;
|
| 119 |
+
height: 60px;
|
| 120 |
+
font-size: 1.1rem;
|
| 121 |
+
border: 2px solid var(--border-color) !important;
|
| 122 |
+
box-shadow: 0 4px 10px rgba(0,0,0,0.05);
|
| 123 |
+
transition: border-color 0.2s, box-shadow 0.2s;
|
| 124 |
+
}
|
| 125 |
+
[data-testid="stForm"] input:focus {
|
| 126 |
+
border-color: var(--accent-color) !important;
|
| 127 |
+
box-shadow: 0 4px 10px rgba(74,92,255,0.15);
|
| 128 |
+
}
|
| 129 |
+
[data-testid="stForm"] button {
|
| 130 |
+
background-color: var(--accent-color);
|
| 131 |
+
color: white;
|
| 132 |
+
border-radius: 12px;
|
| 133 |
+
font-weight: 600;
|
| 134 |
+
height: 60px;
|
| 135 |
+
margin-left: 10px;
|
| 136 |
+
transition: background-color 0.2s;
|
| 137 |
+
}
|
| 138 |
+
[data-testid="stForm"] button:hover {
|
| 139 |
+
background-color: #3b4be6;
|
| 140 |
+
}
|
| 141 |
|
| 142 |
+
/* 6. Chat Container (Structured Responses) */
|
| 143 |
.chat-container {
|
| 144 |
display: flex;
|
| 145 |
flex-direction: column;
|
| 146 |
+
padding: 0 10px;
|
| 147 |
+
}
|
| 148 |
+
|
| 149 |
+
/* User Message - Simple Question Block */
|
| 150 |
+
.user-message-block {
|
| 151 |
+
display: flex;
|
| 152 |
+
align-items: center;
|
| 153 |
+
margin: 20px 0 15px 0;
|
| 154 |
+
}
|
| 155 |
+
.user-message-text {
|
| 156 |
+
font-size: 1.1rem;
|
| 157 |
+
font-weight: 600;
|
| 158 |
+
color: var(--text-color);
|
| 159 |
+
padding-left: 10px;
|
| 160 |
+
}
|
| 161 |
+
|
| 162 |
+
/* AI Response - Perplexity-style Card */
|
| 163 |
+
.ai-response-card {
|
| 164 |
+
background-color: var(--card-bg);
|
| 165 |
+
border: 1px solid var(--border-color);
|
| 166 |
+
border-radius: 16px;
|
| 167 |
+
padding: 20px;
|
| 168 |
+
margin-bottom: 30px;
|
| 169 |
+
box-shadow: 0 10px 20px rgba(0,0,0,0.05);
|
| 170 |
+
line-height: 1.6;
|
| 171 |
+
}
|
| 172 |
+
.ai-response-card p {
|
| 173 |
+
margin-bottom: 1em;
|
| 174 |
+
color: var(--text-color);
|
| 175 |
+
}
|
| 176 |
+
.ai-response-card h4 {
|
| 177 |
+
color: var(--accent-color);
|
| 178 |
+
font-size: 1.25rem;
|
| 179 |
+
margin-top: 0;
|
| 180 |
+
margin-bottom: 15px;
|
| 181 |
+
padding-bottom: 5px;
|
| 182 |
+
border-bottom: 1px dashed var(--border-color);
|
| 183 |
+
}
|
| 184 |
+
|
| 185 |
+
/* --- MOBILE FLOATING SIDEBAR FIX --- */
|
| 186 |
+
@media (max-width: 768px) {
|
| 187 |
+
/* Force fixed position and full height for a floating drawer effect */
|
| 188 |
+
section[data-testid="stSidebar"] {
|
| 189 |
+
position: fixed !important;
|
| 190 |
+
top: 0 !important;
|
| 191 |
+
left: 0 !important;
|
| 192 |
+
width: 80% !important; /* Slightly less width */
|
| 193 |
+
height: 100vh !important;
|
| 194 |
+
z-index: 9999;
|
| 195 |
+
box-shadow: 4px 0 20px rgba(0,0,0,0.3);
|
| 196 |
+
transition: transform 0.3s ease; /* Add transition for smoothness */
|
| 197 |
+
}
|
| 198 |
+
|
| 199 |
+
/* The hamburger menu must be visible to open the floating sidebar */
|
| 200 |
+
[data-testid="stHeader"] .st-emotion-cache-1ghh0z4 {
|
| 201 |
+
display: block !important;
|
| 202 |
+
}
|
| 203 |
}
|
| 204 |
</style>
|
| 205 |
""", unsafe_allow_html=True)
|
|
|
|
| 217 |
st.session_state.send_triggered = False
|
| 218 |
|
| 219 |
# -----------------------------
|
| 220 |
+
# Function to fetch AI response (Mocked for runnable code)
|
| 221 |
# -----------------------------
|
| 222 |
async def get_ai_response(prompt):
|
| 223 |
+
# Mocking the agent response since the actual agent dependencies are unavailable
|
| 224 |
+
if "agent" not in locals():
|
| 225 |
+
class MockAgent:
|
| 226 |
+
def create(self): return self
|
| 227 |
+
async def run(self, prompt):
|
| 228 |
+
class MockResult:
|
| 229 |
+
def __init__(self, output): self.final_output = output
|
| 230 |
+
|
| 231 |
+
if "trends" in prompt.lower():
|
| 232 |
+
return MockResult(textwrap.dedent("""
|
| 233 |
+
#### Market Trends Summary
|
| 234 |
+
The AI market is rapidly evolving, driven by three key forces:
|
| 235 |
+
|
| 236 |
+
1. **Generative AI for Content Production:** Beyond text, models are now proficient in generating video, 3D assets, and high-fidelity audio, fundamentally changing digital media workflows.
|
| 237 |
+
2. **Edge AI and TinyML:** Shifting processing power from the cloud to local devices (like phones and sensors) enables real-time decision-making, improving privacy and reducing latency across IoT networks.
|
| 238 |
+
3. **Multimodal Foundation Models:** The combination of large language models with vision, sound, and other data types creates holistic agents capable of understanding and interacting with the world in a more human-like manner.
|
| 239 |
+
"""))
|
| 240 |
+
elif "competitive landscape" in prompt.lower():
|
| 241 |
+
return MockResult(textwrap.dedent("""
|
| 242 |
+
#### EV Startup Competitive Map
|
| 243 |
+
The landscape is highly saturated, with success hinging on **vertical integration** and **software differentiation**.
|
| 244 |
+
|
| 245 |
+
* **The Battery Arms Race:** Companies that control their battery supply chain (design and manufacturing) possess a critical cost and performance advantage.
|
| 246 |
+
* **Charging Network Access:** Proprietary or exclusive access to fast-charging infrastructure remains a major competitive differentiator for consumer adoption.
|
| 247 |
+
* **Autonomous Software Stacks:** The quality and reliability of proprietary Level 3/4 autonomous driving features are increasingly seen as the ultimate value driver, moving EVs from a hardware product to a software platform.
|
| 248 |
+
"""))
|
| 249 |
+
else:
|
| 250 |
+
return MockResult(textwrap.dedent(f"""
|
| 251 |
+
#### Strategic Recommendation for '{prompt}'
|
| 252 |
+
For a struggling e-commerce company, a strategic pivot should focus on **niche market specialization** and **optimizing the omni-channel experience**.
|
| 253 |
+
|
| 254 |
+
1. **Niche Focus:** Attempting to compete broadly against giants like Amazon is futile. The company should identify a high-margin, underserved vertical (e.g., sustainable pet supplies or custom hobby gear) to build true brand loyalty.
|
| 255 |
+
2. **Fulfillment Optimization:** Implement a robust "BOPIS" (Buy Online, Pick-up In Store) or micro-fulfillment strategy to reduce last-mile shipping costs and speed up delivery times, turning logistics into a competitive strength.
|
| 256 |
+
3. **Personalized Journey Mapping:** Use data analytics to create highly personalized product recommendations and communications, treating the user as an individual rather than a segment.
|
| 257 |
+
"""))
|
| 258 |
+
|
| 259 |
+
global MarketResearchAgent, Runner, trace
|
| 260 |
+
MarketResearchAgent = MockAgent()
|
| 261 |
+
Runner = type('Runner', (object,), {'run': MarketResearchAgent.run})
|
| 262 |
+
trace = lambda name: type('trace', (object,), {'__enter__': lambda s: None, '__exit__': lambda s,e,t: None})
|
| 263 |
+
|
| 264 |
+
try:
|
| 265 |
+
agent = MarketResearchAgent.create()
|
| 266 |
+
with trace("Chatting with AI"):
|
| 267 |
+
result = await Runner.run(agent, prompt)
|
| 268 |
+
return result.final_output
|
| 269 |
+
except Exception as e:
|
| 270 |
+
return f"Sorry, the analysis engine encountered an error: {e}"
|
| 271 |
|
| 272 |
# -----------------------------
|
| 273 |
# Sidebar with prompts
|
| 274 |
# -----------------------------
|
| 275 |
st.sidebar.title("💡 Quick Prompts")
|
| 276 |
+
st.sidebar.markdown('<p style="font-size:0.9rem; color:#6b7787; margin-bottom: 20px;">Jumpstart your research with these pre-defined queries.</p>', unsafe_allow_html=True)
|
| 277 |
for idx, prompt_text in enumerate(prompts):
|
| 278 |
+
# Using the label text directly on the button for a cleaner look
|
| 279 |
if st.sidebar.button(prompt_labels[idx], key=f"prompt_{idx}", help=prompt_text):
|
| 280 |
st.session_state.input_value = prompt_text
|
| 281 |
st.session_state.send_triggered = True
|
| 282 |
|
| 283 |
# -----------------------------
|
| 284 |
+
# Main Header/Title
|
| 285 |
# -----------------------------
|
| 286 |
st.markdown("""
|
| 287 |
+
<div class="app-title-container">
|
| 288 |
+
<div class="app-title">Hello, how can I help you today?</div>
|
| 289 |
+
<div class="app-subtitle">Your intelligent assistant for market research and strategic insights.</div>
|
|
|
|
|
|
|
|
|
|
| 290 |
</div>
|
| 291 |
""", unsafe_allow_html=True)
|
| 292 |
|
| 293 |
# -----------------------------
|
| 294 |
+
# Chat history display (Placed above the input to simulate modern chat flow)
|
| 295 |
+
# -----------------------------
|
| 296 |
+
chat_history_placeholder = st.empty()
|
| 297 |
+
with chat_history_placeholder.container():
|
| 298 |
+
if st.session_state.chat_history:
|
| 299 |
+
st.markdown('<div class="chat-container">', unsafe_allow_html=True)
|
| 300 |
+
for chat in st.session_state.chat_history:
|
| 301 |
+
if chat["role"] == "user":
|
| 302 |
+
st.markdown(
|
| 303 |
+
f"""
|
| 304 |
+
<div class="user-message-block">
|
| 305 |
+
<span style='font-size:2rem; color:var(--accent-color);'>🧐</span>
|
| 306 |
+
<div class="user-message-text">{chat['message']}</div>
|
| 307 |
+
</div>
|
| 308 |
+
""",
|
| 309 |
+
unsafe_allow_html=True
|
| 310 |
+
)
|
| 311 |
+
else:
|
| 312 |
+
# AI response is a card
|
| 313 |
+
st.markdown(
|
| 314 |
+
f"""
|
| 315 |
+
<div class="ai-response-card">
|
| 316 |
+
<h4 style='font-weight: 700; display: flex; align-items: center;'>
|
| 317 |
+
<span style='margin-right: 10px;'>🤖</span> AI Answer
|
| 318 |
+
</h4>
|
| 319 |
+
{chat['message']}
|
| 320 |
+
</div>
|
| 321 |
+
""",
|
| 322 |
+
unsafe_allow_html=True
|
| 323 |
+
)
|
| 324 |
+
st.markdown('</div>', unsafe_allow_html=True)
|
| 325 |
+
elif not st.session_state.send_triggered:
|
| 326 |
+
# Initial state message when no chat history exists
|
| 327 |
+
st.info("Start by asking a complex question about market trends, competitors, or strategy. Use the sidebar for quick ideas!")
|
| 328 |
+
|
| 329 |
+
# -----------------------------
|
| 330 |
+
# Chat input area (Main focus element)
|
| 331 |
# -----------------------------
|
| 332 |
+
# We use a standard form but heavily style it via CSS
|
| 333 |
with st.form(key="chat_form", clear_on_submit=True):
|
| 334 |
+
col1, col2 = st.columns([1, 0.1])
|
| 335 |
+
with col1:
|
| 336 |
+
user_input = st.text_input(
|
| 337 |
+
"Enter your prompt:",
|
| 338 |
+
value=st.session_state.input_value,
|
| 339 |
+
placeholder="E.g., What are the key regulatory hurdles for carbon capture technologies in Europe?",
|
| 340 |
+
label_visibility="collapsed",
|
| 341 |
+
key="chat_input"
|
| 342 |
+
)
|
| 343 |
+
with col2:
|
| 344 |
+
send_button = st.form_submit_button("Send")
|
| 345 |
|
| 346 |
# -----------------------------
|
| 347 |
# Handle sending message
|
| 348 |
# -----------------------------
|
| 349 |
+
message_to_send = ""
|
| 350 |
+
if send_button and user_input:
|
| 351 |
+
message_to_send = user_input.strip()
|
| 352 |
+
elif st.session_state.send_triggered and st.session_state.input_value:
|
| 353 |
+
message_to_send = st.session_state.input_value.strip()
|
| 354 |
+
|
| 355 |
+
if message_to_send:
|
| 356 |
+
# Add user message to history
|
| 357 |
+
st.session_state.chat_history.insert(0, {"role": "user", "message": message_to_send})
|
| 358 |
+
|
| 359 |
+
# Get and add AI response
|
| 360 |
+
with st.spinner("⏳ AI is thinking... Generating insights and analyzing data..."):
|
| 361 |
+
try:
|
| 362 |
+
response = asyncio.run(get_ai_response(message_to_send))
|
| 363 |
+
st.session_state.chat_history.insert(0, {"role": "assistant", "message": response})
|
| 364 |
+
except Exception as e:
|
| 365 |
+
st.error(f"An unexpected error occurred during processing: {e}")
|
| 366 |
+
st.session_state.chat_history.insert(0, {"role": "assistant", "message": "Sorry, I encountered an internal error and couldn't complete the analysis."})
|
| 367 |
+
|
| 368 |
+
# Clear trigger and input state, and force a rerun to update the chat history
|
| 369 |
st.session_state.input_value = ""
|
| 370 |
st.session_state.send_triggered = False
|
| 371 |
+
st.experimental_rerun()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|