Update app.py

app.py

@@ -1,37 +1,36 @@
 import os
 import streamlit as st
-from langchain.chat_models import ChatOpenAI
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
-import openai
-import plotly.graph_objects as go
-from xhtml2pdf import pisa
+from langchain_community.chat_models import ChatOpenAI
+from openai import OpenAIError, RateLimitError, InvalidRequestError
+import matplotlib.pyplot as plt
+from fpdf import FPDF
 import tempfile
 
-# --- Configuration ---
-st.set_page_config(page_title="🔍 Competitive Analysis Tool")
-st.title("🔍 Competitive Analysis Tool")
+# Set OpenAI API Key from Hugging Face secrets
+os.environ["OPENAI_API_KEY"] = st.secrets["OPENAI_API_KEY"]
 
-# --- 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()
+# Streamlit UI
+st.set_page_config(page_title="🆚 Competitive Analysis Tool", layout="wide")
+st.title("🆚 Competitive Analysis Tool")
+st.markdown("Compare two products or services with **Feature Comparison**, **SWOT Analysis**, and a **Summary**.")
 
-# --- 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()
+# --- Inputs ---
+product1 = st.text_area("🔹 Enter details about Product/Service 1", height=150)
+product2 = st.text_area("🔹 Enter details about Product/Service 2", height=150)
 
 # --- Prompt Template ---
 template = """
-Compare the following two products/services: {product1} and {product2}.
-Provide the following:
-1. A feature-by-feature comparison.
-2. SWOT analysis for each product/service.
-3. A comparative summary highlighting key differentiators.
+Compare the following two products or services:
+[Product 1]: {product1}
+[Product 2]: {product2}
+
+1. Feature-by-feature comparison in a table format.
+2. SWOT analysis for each product.
+3. Summary highlighting key differentiators.
+
+Be concise, clear, and professional.
 """
 
 prompt = PromptTemplate(
@@ -39,60 +38,72 @@ prompt = PromptTemplate(
     template=template,
 )
 
+# --- Chain ---
+llm = ChatOpenAI(temperature=0.2, model_name="gpt-3.5-turbo")
 comparison_chain = LLMChain(llm=llm, prompt=prompt)
 
-# --- 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
-
-# --- Chart Drawing (Static Example) ---
+# --- PDF Converter ---
+def convert_to_pdf(html_text):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.set_font("Arial", size=12)
+    for line in html_text.split('\n'):
+        pdf.multi_cell(0, 10, line)
+    temp_path = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
+    pdf.output(temp_path)
+    return temp_path
+
+# --- Chart Drawer (dummy feature count for visual demo) ---
 def draw_feature_chart():
-    features = ["Storage", "Collaboration", "Sync Speed", "Security"]
-    scores1 = [8, 9, 7, 8]
-    scores2 = [7, 6, 9, 7]
-
-    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)
-
-# --- 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")
-
-# --- On Submit ---
-if submitted:
-    if not product1.strip() or not product2.strip():
-        st.warning("Please enter descriptions for both products.")
-    else:
-        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)}")
-
-            except openai.error.OpenAIError as e:
-                st.error(f"💥 OpenAI Error: {str(e)}")
+    features = ["Ease of Use", "Pricing", "Customization", "Support", "Scalability"]
+    product1_scores = [7, 6, 8, 9, 7]
+    product2_scores = [8, 7, 6, 8, 9]
+
+    fig, ax = plt.subplots()
+    bar_width = 0.35
+    x = range(len(features))
+
+    ax.bar(x, product1_scores, width=bar_width, label='Product 1')
+    ax.bar([p + bar_width for p in x], product2_scores, width=bar_width, label='Product 2')
+
+    ax.set_xlabel('Features')
+    ax.set_ylabel('Score')
+    ax.set_title('Feature Comparison Chart')
+    ax.set_xticks([p + bar_width/2 for p in x])
+    ax.set_xticklabels(features)
+    ax.legend()
+
+    st.pyplot(fig)
+
+# --- Generate Output ---
+if st.button("⚖️ Compare Now") and product1 and product2:
+    with st.spinner("Analyzing with LLM..."):
+        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"Competitive Analysis\n\n{result}")
+            with open(pdf_path, "rb") as f:
+                st.download_button("📄 Download PDF Report", f, file_name="analysis_report.pdf")
+
+        except RateLimitError:
+            st.error("🚫 You exceeded your OpenAI quota. Check https://platform.openai.com/account/usage")
+
+        except InvalidRequestError as e:
+            st.error(f"❌ Invalid request: {str(e)}")
+
+        except OpenAIError as e:
+            st.error(f"💥 OpenAI Error: {str(e)}")
+
+# Optional Footer
+st.markdown("---")
+st.markdown("Built with 💡 using Langchain, Streamlit, and OpenAI.")