Update app.py

app.py

@@ -1,55 +1,98 @@
+import os
 import streamlit as st
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
-import os
+import openai
+import plotly.graph_objects as go
+from xhtml2pdf import pisa
+import tempfile
 
-# Set your OpenAI key
-openai_api_key = os.getenv("OPENAI_API_KEY")
-os.environ["OPENAI_API_KEY"] = openai_api_key
+# --- Configuration ---
+st.set_page_config(page_title="🔍 Competitive Analysis Tool")
+st.title("🔍 Competitive Analysis Tool")
 
-# Prompt template for comparison
-prompt_template = PromptTemplate(
-    input_variables=["product1", "product2"],
-    template="""
-Compare the following two products/services:
+# --- OpenAI Key ---
+openai_api_key = os.getenv("OPENAI_API_KEY")
+if not openai_api_key:
+    st.error("❌ OPENAI_API_KEY not found in environment variables.")
+    st.stop()
 
-Product 1: {product1}
-Product 2: {product2}
+# --- Initialize LLM ---
+try:
+    llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0, openai_api_key=openai_api_key)
+except openai.error.OpenAIError as e:
+    st.error(f"Failed to initialize OpenAI: {str(e)}")
+    st.stop()
 
-Please provide the following:
+# --- Prompt Template ---
+template = """
+Compare the following two products/services: {product1} and {product2}.
+Provide the following:
 1. A feature-by-feature comparison.
-2. A SWOT (Strengths, Weaknesses, Opportunities, Threats) analysis for each.
+2. SWOT analysis for each product/service.
 3. A comparative summary highlighting key differentiators.
 """
+
+prompt = PromptTemplate(
+    input_variables=["product1", "product2"],
+    template=template,
 )
 
-# Initialize LLM and chain
-llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0)
-comparison_chain = LLMChain(llm=llm, prompt=prompt_template)
+comparison_chain = LLMChain(llm=llm, prompt=prompt)
 
-# Streamlit UI
-st.set_page_config(page_title="Competitive Analysis Tool", layout="wide")
-st.title("🔍 Competitive Analysis Tool")
+# --- PDF Conversion ---
+def convert_to_pdf(text: str):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
+        pisa.CreatePDF(text, dest=tmp_file)
+        return tmp_file.name
 
-st.markdown("Compare two products or services and get a full SWOT analysis and feature comparison.")
+# --- Chart Drawing (Static Example) ---
+def draw_feature_chart():
+    features = ["Storage", "Collaboration", "Sync Speed", "Security"]
+    scores1 = [8, 9, 7, 8]
+    scores2 = [7, 6, 9, 7]
 
-# Input fields
-product1 = st.text_area("📝 Enter details of Product/Service 1", height=200)
-product2 = st.text_area("📝 Enter details of Product/Service 2", height=200)
+    fig = go.Figure(data=[
+        go.Bar(name='Product 1', x=features, y=scores1),
+        go.Bar(name='Product 2', x=features, y=scores2)
+    ])
+    fig.update_layout(barmode='group', title="📊 Feature Comparison")
+    st.plotly_chart(fig, use_container_width=True)
 
-# Button
-if st.button("Compare"):
-    if product1.strip() and product2.strip():
-        with st.spinner("Generating analysis..."):
-            result = comparison_chain.run(product1=product1, product2=product2)
+# --- UI Input ---
+with st.form("product_form"):
+    product1 = st.text_area("Enter details for Product/Service 1", height=150, placeholder="e.g., Google Drive: Cloud storage with real-time collaboration.")
+    product2 = st.text_area("Enter details for Product/Service 2", height=150, placeholder="e.g., Dropbox: File sharing and sync service with smart folders.")
+    submitted = st.form_submit_button("Compare")
 
-        # Display result in separate sections
-        st.subheader("📊 Feature Comparison & SWOT Analysis")
-        st.markdown(result)
+# --- On Submit ---
+if submitted:
+    if not product1.strip() or not product2.strip():
+        st.warning("Please enter descriptions for both products.")
     else:
-        st.warning("Please enter details for both products/services.")
+        with st.spinner("🔍 Generating analysis..."):
+            try:
+                result = comparison_chain.run(product1=product1, product2=product2)
+
+                # --- Show Results ---
+                st.subheader("📋 Full Analysis")
+                st.markdown(result)
+
+                # --- Show Chart ---
+                st.subheader("📊 Feature Comparison (Chart)")
+                draw_feature_chart()
+
+                # --- Download PDF ---
+                pdf_path = convert_to_pdf(f"<h1>Competitive Analysis</h1><pre>{result}</pre>")
+                with open(pdf_path, "rb") as f:
+                    st.download_button("📄 Download PDF Report", f, file_name="analysis_report.pdf")
+
+            except openai.error.RateLimitError:
+                st.error("🚫 You exceeded your OpenAI quota. Check https://platform.openai.com/account/usage")
+
+            except openai.error.InvalidRequestError as e:
+                st.error(f"❌ Invalid request: {str(e)}")
 
-# Footer
-st.markdown("---")
-st.caption("Powered by OpenAI + Langchain + Streamlit")
+            except openai.error.OpenAIError as e:
+                st.error(f"💥 OpenAI Error: {str(e)}")