Update app.py

app.py

@@ -4,8 +4,9 @@ import os
 from fpdf import FPDF
 from io import BytesIO
 import textwrap
+import re
 
-# Set page layout and styling
+# Set page layout and dark theme styling
 st.set_page_config(page_title="Pro Competitive Analysis", layout="centered")
 st.markdown("""
 <style>
@@ -37,33 +38,36 @@ button:hover {
 
 # App header
 st.title("💼 Competitive Analysis Pro")
-st.markdown("Generate expert-level comparisons of two products or services, including SWOT, business recommendations, and more.")
+st.markdown("Generate expert-level comparisons between two products or services, with SWOT, insights, and business recommendations.")
 
-# Set Hugging Face API key
+# Set Hugging Face API key (securely via Secrets)
 os.environ["HUGGINGFACEHUB_API_TOKEN"] = st.secrets["HF_TOKEN"]
 
-# Initialize the LLM
+# LLM Setup
 llm = HuggingFaceHub(
     repo_id="tiiuae/falcon-7b-instruct",
     model_kwargs={"temperature": 0.7, "max_new_tokens": 1024}
 )
 
-# User inputs
+# Input Fields
 product1 = st.text_area("🧩 Product/Service 1", height=200, placeholder="e.g., iPhone 13")
 product2 = st.text_area("🧩 Product/Service 2", height=200, placeholder="e.g., iPhone 14")
 
-# On click
+# Button logic
 if st.button("🔍 Run Competitive Analysis", use_container_width=True):
     if not product1 or not product2:
-        st.warning("Please enter descriptions for both products.")
+        st.warning("Please enter both product descriptions.")
     else:
-        # Prompt for the model
+        # LLM prompt (not printed)
         system_instruction = """
 You are a professional market analyst trained to evaluate two competing products or services.
 
 Write an in-depth comparative analysis in **markdown format**, covering:
 
 1. Feature-by-Feature Comparison:
+
+- For both Product 1 and Product 2 separately
+
 - Hardware specifications
 - Design and display
 - Performance (CPU, GPU, etc.)
@@ -87,68 +91,50 @@ Write an in-depth comparative analysis in **markdown format**, covering:
 
 ⚠️ Do NOT include or reference these instructions in your output. Return only the clean markdown report.
 """
+        user_input = f"\n\nProduct 1:\n{product1}\n\nProduct 2:\n{product2}"
+        final_input = system_instruction.strip() + user_input
 
-        user_input = f"""
-Product 1:
-{product1}
-
-Product 2:
-{product2}
-"""
-        final_input = system_instruction.strip() + "\n\n" + user_input.strip()
-
-        # Run the LLM
         with st.spinner("🧠 Generating detailed analysis..."):
             result = llm(final_input)
 
-        # Display markdown result
         st.markdown("### 📊 Expert Comparison")
         st.markdown(result)
 
-        # Sanitize and wrap text for PDF
-        from fpdf import FPDF
-from io import BytesIO
-import textwrap
-import re
-
-# Clean markdown
-clean_text = result.replace("**", "").replace("*", "").replace("`", "").replace("#", "").replace("•", "-")
-
-# Init PDF
-pdf = FPDF()
-pdf.add_page()
-pdf.set_font("Arial", size=12)
-pdf.set_auto_page_break(auto=True, margin=15)
-
-def safe_line_split(text, width=100):
-    # Split long non-space strings if needed
-    words = re.findall(r'.{1,%d}' % width, text) if len(text) > width else [text]
-    return words
-
-for line in clean_text.split("\n"):
-    line = line.strip()
-    if not line:
-        pdf.ln(5)
-        continue
-    try:
-        wrapped_lines = textwrap.wrap(line, width=100)
-        if not wrapped_lines:
-            wrapped_lines = safe_line_split(line)
-        for wl in wrapped_lines:
-            pdf.multi_cell(0, 10, wl)
-    except Exception as e:
-        print(f"⚠️ Skipped line (too long or invalid): {line[:50]}...")
-
-# Output stream
-pdf_output = BytesIO()
-pdf.output(pdf_output)
-pdf_output.seek(0)
-
-# Download
-st.download_button(
-    label="📥 Download Report as PDF",
-    data=pdf_output,
-    file_name="competitive_analysis_report.pdf",
-    mime="application/pdf",
-    use_container_width=True
-)
+        # Clean markdown & sanitize output
+        clean_text = result.replace("**", "").replace("*", "").replace("`", "").replace("#", "").replace("•", "-")
+
+        # PDF Generation (safe against long unbreakable lines)
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Arial", size=12)
+        pdf.set_auto_page_break(auto=True, margin=15)
+
+        def safe_line_split(text, width=100):
+            return re.findall(r'.{1,%d}' % width, text) if len(text) > width else [text]
+
+        for line in clean_text.split("\n"):
+            line = line.strip()
+            if not line:
+                pdf.ln(5)
+                continue
+            try:
+                wrapped_lines = textwrap.wrap(line, width=100)
+                if not wrapped_lines:
+                    wrapped_lines = safe_line_split(line)
+                for wl in wrapped_lines:
+                    pdf.multi_cell(0, 10, wl)
+            except Exception as e:
+                print(f"⚠️ Skipped line due to error: {line[:50]}...")
+
+        pdf_output = BytesIO()
+        pdf.output(pdf_output)
+        pdf_output.seek(0)
+
+        # Streamlit download button
+        st.download_button(
+            label="📥 Download Report as PDF",
+            data=pdf_output,
+            file_name="competitive_analysis_report.pdf",
+            mime="application/pdf",
+            use_container_width=True
+        )