Update app.py

app.py

@@ -3,107 +3,102 @@ import streamlit as st
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain_community.chat_models import ChatOpenAI
-from openai.error import OpenAIError, RateLimitError, InvalidRequestError
-import matplotlib.pyplot as plt
-from fpdf import FPDF
+from matplotlib import pyplot as plt
 import tempfile
+from fpdf import FPDF
+import openai
 
-# Set OpenAI API Key from Hugging Face secrets
+# 🔐 Set your OpenAI API key from Hugging Face secret
 os.environ["OPENAI_API_KEY"] = st.secrets["OPENAI_API_KEY"]
 
-# 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**.")
-
-# --- 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
+prompt_template = PromptTemplate(
+    input_variables=["product1", "product2"],
+    template="""
+Compare the following two products/services:
 
-# --- Prompt Template ---
-template = """
-Compare the following two products or services:
-[Product 1]: {product1}
-[Product 2]: {product2}
+Product 1: {product1}
 
-1. Feature-by-feature comparison in a table format.
-2. SWOT analysis for each product.
-3. Summary highlighting key differentiators.
+Product 2: {product2}
 
-Be concise, clear, and professional.
+1. Provide a detailed feature-by-feature comparison.
+2. Perform a SWOT analysis for each.
+3. Conclude with a comparative summary highlighting key differentiators.
 """
-
-prompt = PromptTemplate(
-    input_variables=["product1", "product2"],
-    template=template,
 )
 
-# --- Chain ---
-llm = ChatOpenAI(temperature=0.2, model_name="gpt-3.5-turbo")
-comparison_chain = LLMChain(llm=llm, prompt=prompt)
+# 🤖 LLM Model Setup
+llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.3)
+comparison_chain = LLMChain(llm=llm, prompt=prompt_template)
 
-# --- PDF Converter ---
+# 📊 Helper: Draw dummy chart for features
+def draw_feature_chart():
+    labels = ['Price', 'Ease of Use', 'Performance', 'Support', 'Customization']
+    product1_scores = [7, 8, 6, 9, 7]
+    product2_scores = [6, 7, 9, 8, 6]
+
+    x = range(len(labels))
+    plt.figure(figsize=(8, 4))
+    plt.bar(x, product1_scores, width=0.4, label="Product 1", align='center')
+    plt.bar([i + 0.4 for i in x], product2_scores, width=0.4, label="Product 2", align='center')
+    plt.xticks([i + 0.2 for i in x], labels)
+    plt.ylabel("Rating (out of 10)")
+    plt.title("Feature Comparison")
+    plt.legend()
+    st.pyplot(plt)
+
+# 📝 Helper: Convert text to PDF
 def convert_to_pdf(text):
     pdf = FPDF()
-    pdf.add_page()
     pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.add_page()
     pdf.set_font("Arial", size=12)
+
     for line in 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 = ["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')
+    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    pdf.output(temp_pdf.name)
+    return temp_pdf.name
 
-    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()
+# 🚀 Streamlit UI
+st.set_page_config(page_title="🔍 Competitive Analysis Tool", layout="wide")
+st.title("🤖 Competitive Analysis Tool using LLMs")
+st.write("Compare two products or services with a detailed feature comparison, SWOT analysis, and summary.")
 
-    st.pyplot(fig)
+# ✍️ User Inputs
+product1 = st.text_area("🅰️ Enter details about Product/Service 1", height=200)
+product2 = st.text_area("🅱️ Enter details about Product/Service 2", height=200)
 
-# --- 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)
+if st.button("🔍 Compare Now"):
+    if not product1 or not product2:
+        st.warning("⚠️ Please enter details for both products/services.")
+    else:
+        with st.spinner("Analyzing... Please wait."):
+            try:
+                result = comparison_chain.run(product1=product1, product2=product2)
 
-            # --- Show Results ---
-            st.subheader("📋 Full Analysis")
-            st.markdown(result)
+                # Display results
+                st.subheader("📋 Full Analysis")
+                st.markdown(result)
 
-            # --- Show Chart ---
-            st.subheader("📊 Feature Comparison (Chart)")
-            draw_feature_chart()
+                # 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")
+                # PDF Download
+                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 openai.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 openai.InvalidRequestError as e:
+                st.error(f"❌ Invalid request: {str(e)}")
 
-        except OpenAIError as e:
-            st.error(f"💥 OpenAI Error: {str(e)}")
+            except openai.OpenAIError as e:
+                st.error(f"💥 OpenAI Error: {str(e)}")
 
-# Optional Footer
-st.markdown("---")
-st.markdown("Built with 💡 using Langchain, Streamlit, and OpenAI.")
+            except Exception as e:
+                st.error(f"⚠️ Unexpected Error: {str(e)}")