Create funtions.py

funtions.py

@@ -0,0 +1,169 @@
+import streamlit as st
+from docx import Document
+import re
+import io
+import os
+from fpdf import FPDF
+from groq import Groq
+from exa_py import Exa
+from dotenv import load_dotenv
+from retrying import retry
+from funtions import *
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Declare the exa search API
+exa = Exa(api_key=os.getenv("EXA_API_KEY"))
+
+# Define your API Model and key
+client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+utilized_model = "llama3-70b-8192"
+
+# Functions for the Exa Search content & Parameters for Highlights search
+highlights_options = {
+    "num_sentences": 7,  # Length of highlights
+    "highlights_per_url": 1,  # Get the best highlight for each URL
+}
+
+@retry(wait_exponential_multiplier=1000, wait_exponential_max=10000, stop_max_attempt_number=5)
+def call_llm(prompt):
+    search_response = exa.search_and_contents(query=prompt, highlights=highlights_options, num_results=3, use_autoprompt=True)
+    info = [sr.highlights[0] for sr in search_response.results]
+    
+    system_prompt = "You are a Business proposal generator. Read the provided contexts and, if relevant, use them to answer the user's question."
+    user_prompt = f"Sources: {info}\nQuestion: {prompt}"
+    
+    completion = client.chat.completions.create(
+        model=utilized_model,
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt},
+        ]
+    )
+    return completion.choices[0].message.content
+
+def strip_md(text):
+    text = text.replace("**", "").replace("*", "").replace("#", "")
+    return re.sub(r'([!*_=~-])', r'\\\1', text)
+
+def create_document():
+    doc = Document()
+    doc.add_heading("Business Proposal", 0)
+    return doc
+
+def add_section_to_doc(doc, section_name, section_content):
+    section_content = strip_md(section_content)
+    section_content = section_content.replace("\\", "")  # Remove backslashes
+    doc.add_heading(section_name, level=1)
+    doc.add_paragraph(section_content)
+    return doc
+
+def get_docx_bytes(doc):
+    doc_io = io.BytesIO()
+    doc.save(doc_io)
+    doc_io.seek(0)
+    return doc_io
+
+def collect_basic_info():
+    st.title("Business Proposal Generator")
+
+    # Basic Company Information
+    company_name = st.text_input("Company Name")
+    industry = st.text_input("Industry")
+    location = st.text_input("Location")
+    mission = st.text_area("Mission Statement")
+    vision = st.text_area("Vision Statement")
+    products_services = st.text_area("Description of Products/Services")
+    target_market = st.text_area("Customer Segments")
+    value_proposition = st.text_area("Competitive Advantage")
+    promotional_strategy = st.text_area("Promotional Strategy")
+    current_revenue = st.number_input("Current Revenue (R)", min_value=0.0, format="%f")
+    current_expenses = st.number_input("Current Expenses (R)", min_value=0.0, format="%f")
+    funding_requirements = st.text_area("Funding Requirements")
+    management_team = st.text_area("Key Personnel")
+    company_structure = st.text_area("Company Structure")
+    goals_objectives = st.text_area("Short-term and Long-term Goals")
+    operational_strategy = st.text_area("Operational Strategy")
+    market_overview = st.text_area("Market Overview")
+
+    if st.button('Submit'):
+        # Collect data
+        data = {
+            "company_name": company_name,
+            "industry": industry,
+            "location": location,
+            "mission": mission,
+            "vision": vision,
+            "products_services": products_services,
+            "target_market": target_market,
+            "value_proposition": value_proposition,
+            "current_revenue": current_revenue,
+            "current_expenses": current_expenses,
+            "funding_requirements": funding_requirements,
+            "management_team": management_team,
+            "company_structure": company_structure,
+            "goals_objectives": goals_objectives,
+            "operational_strategy": operational_strategy,
+            "market_overview": market_overview,
+            "promotional_strategy": promotional_strategy
+        }
+
+        # Create a new document
+        doc = create_document()
+
+        # Process and update document with each section
+        sections_to_process = [
+            ("Executive Summary", generate_executive_summary),
+            ("Mission Statement", generate_mission),
+            ("Vision Statement", generate_vision),
+            ("Objectives", generate_objectives),
+            ("Core Values", generate_core_values),
+            ("Business Description Analysis", generate_business_description),
+            ("Company Location", generate_company_location),
+            ("Products", generate_products),
+            ("Ownership", generate_ownership),
+            ("Company Structure", generate_company_structure),
+            ("Management Profiles", generate_management_profiles),
+            ("Operational Strategy", generate_operational_strategy),
+            ("Marketing Mix Strategy", generate_marketing_mix),
+            ("Promotional Strategy", generate_promotional_strategy),
+            ("Market Demand Analysis", analyze_demand),
+            ("Market Segment Analysis", segment_market),
+            ("Competitor Analysis", analyze_competitors),
+            ("Porter's Five Forces Analysis", perform_porters_five_forces),
+            ("Industry Analysis", analyze_industry_accommodation),
+            ("Major Player Analysis", list_major_players),
+            ("Business Sub Sector Analysis", analyze_business_sub_sector),
+            ("SWOT Analysis", generate_swot_analysis),
+            ("Funding Request", generate_funding_request),
+            ("Financing & Bank Loan Amortization", create_financing_plan),
+            ("Income Statement Analysis", generate_pro_forma_income_statement),
+            ("Revenue Expense Analysis", predict_revenue_expenses),
+            ("Montly Cash Flow Analysis", generate_monthly_cash_flow),
+            ("Pro Forma Annual Cash Flow Analysis", generate_pro_forma_annual_cash_flow),
+            ("Pro Forma Balance Sheet Analysis", generate_pro_forma_balance_sheet),
+            ("Break-Even Analysis", perform_break_even_analysis),
+            ("Payback Period Analysis", calculate_payback_period),
+            ("Financial Graphs Analysis", generate_financial_graphs),
+            ("Risk Mitigations Analysis", identify_risks_mitigations)
+        ]
+
+        for section_name, generate_prompt_func in sections_to_process:
+            prompt = generate_prompt_func(data)
+            section_content = call_llm(prompt)
+            st.subheader(section_name)
+            st.write(section_content)
+            
+            # Update document and create download link
+            doc = add_section_to_doc(doc, section_name, section_content)
+            doc_bytes = get_docx_bytes(doc)
+
+            st.download_button(
+                label=f"Download {section_name} as DOCX",
+                data=doc_bytes,
+                file_name=f"{section_name.replace(' ', '_').lower()}.docx",
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+            )
+
+collect_basic_info()