first commit

.gitattributes

@@ -0,0 +1,3 @@
+*.png filter=lfs diff=lfs merge=lfs -text
+*.PNG filter=lfs diff=lfs merge=lfs -text
+*.docx filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+.streamlit/secrets.toml
+__pycache__

Files/ZEGA AI Capabilities Overview for Clients (1).docx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e9846b0d831376616c6c92b7e65455a5e930e5a6b4126d36746e9876ffb2d24
+size 199448

Files/ZEGA AI Document Capabilities Overview for Clients (1).docx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7651fbd13a38467e77598d01b7e0c893ff243810f1c958e6f6163029d34ded94
+size 9748

app.py

@@ -0,0 +1,80 @@
+import streamlit as st
+import os
+from query_chat import GeminiQanA
+from docx import Document
+
+# Function to read text from a Word document
+def extract_text_from_docx(file_path):
+    doc = Document(file_path)
+    return " ".join([para.text.strip() for para in doc.paragraphs if para.text.strip()])
+
+# Streamlit App Configuration
+st.set_page_config(page_title="Zega AI Sales Agent", page_icon="🤖", layout="centered")
+
+# Sidebar with branding
+st.sidebar.image("zega_logo.PNG", width=300)
+st.sidebar.markdown("### Welcome to Zega AI Sales Agent!")
+st.sidebar.markdown("Ask anything about our team's capabilities and projects.")
+
+# Load API key safely
+if "GOOGLE_API_KEY" in st.secrets:
+    os.environ["GOOGLE_API_KEY"] = st.secrets["GOOGLE_API_KEY"]
+else:
+    st.error("API key missing! Please set up your Google API key in Streamlit secrets.")
+
+# Load documents
+with st.spinner("Loading project information..."):
+    doc1_text = extract_text_from_docx("Files/ZEGA AI Capabilities Overview for Clients (1).docx")
+    doc2_text = extract_text_from_docx("Files/ZEGA AI Document Capabilities Overview for Clients (1).docx")
+
+# Initialize chatbot
+chatbot = GeminiQanA(doc1_text, doc2_text)
+
+# Initialize chat session
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Chat UI
+st.title("📄 Zega AI Sales Agent")
+
+# Display chat history
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# User Input
+question = st.text_area("Ask a question about Zega AI:", height=100)
+
+# Chat Actions
+col1, col2 = st.columns([3, 1])
+with col1:
+    ask_button = st.button("💬 Ask AI")
+with col2:
+    clear_button = st.button("🗑️ Clear Chat")
+
+# Clear chat history
+if clear_button:
+    st.session_state.messages = []
+    st.rerun()
+
+# Handle user input
+if ask_button and question:
+    # Append user question
+    st.session_state.messages.append({"role": "user", "content": question})
+    
+    # Display user message
+    with st.chat_message("user"):
+        st.markdown(question)
+
+    # Generate AI response
+    with st.spinner("💡 Thinking..."):
+        answer = chatbot.answer_question(question)
+
+    # Append AI response
+    st.session_state.messages.append({"role": "assistant", "content": answer})
+
+    # Display AI response
+    with st.chat_message("assistant"):
+        st.markdown(answer)
+        
+    st.rerun()

query_chat.py

@@ -0,0 +1,73 @@
+import google.generativeai as genai
+import os
+
+class GeminiQanA:
+    def __init__(self,text1:str='',text2:str=''):
+        """Initializes the Gemini question answerer by loading the model."""
+        self.api_key = os.getenv("GOOGLE_API_KEY")
+        genai.configure(api_key=self.api_key)
+        self.model = self._load_model(text1,text2)
+
+    def _load_model(self,text1:str,text2:str):
+        """Loads the generative AI model with a system instruction."""
+        final_prompt = f'''Role:
+You are a sales agent responsible for assisting customers by answering questions about our team’s capabilities and the projects we offer. You have access to two brochures that detail the available projects and their features. Your goal is to provide accurate and honest responses based solely on the information within these brochures.
+
+Guidelines for Responses:
+1. Accuracy & Honesty
+-Only provide responses based on the brochures.
+-Do not overstate or exaggerate the capabilities of the team.
+-If information is not available in the brochures, do not speculate—politely inform the customer that the requested details are not available.
+2. Answering Questions About the Team’s Capabilities & Projects
+-When a customer asks about what our team can do, provide information only from the brochures.
+-If asked about past projects, refer only to those explicitly mentioned in the brochures.
+-If the customer asks for additional details not found in the brochures, politely inform them that you can only share the information available.
+3. Providing Solutions to Customer Problems
+-If a customer presents a problem, check if a project in the brochures provides a direct solution.
+-If a matching project exists, explain how it can address their problem.
+-If an alternative but related project exists, suggest it as a partial solution, explaining its limitations.
+-If no project can help, politely state that no suitable solution is available.
+4. What Not to Do
+-Do not create new information or assume additional capabilities.
+-Do not make guarantees beyond what is stated in the brochures.
+-Do not offer speculative solutions that are not explicitly supported by the documents.
+Example Interactions:
+Scenario 1: Customer Asks About a Team Capability
+✅ Customer: "Does your team specialize in AI-powered automation?"
+✅ Agent: "According to our brochure, our team specializes in [list relevant capabilities]. While AI-powered automation is not specifically mentioned, we do offer [related project] which may align with your needs."
+
+Scenario 2: Customer Has a Specific Problem
+✅ Customer: "I need a system to manage logistics for my e-commerce business. Do you have a solution?"
+✅ Agent: "Yes, we offer [Project Name], which is designed for logistics management. It provides [brief relevant details from the brochure]. Would you like more information on its features?"
+
+Scenario 3: No Matching Solution Available
+✅ Customer: "Do you have a tool for automating customer sentiment analysis?"
+✅ Agent: "I'm sorry, but our current projects do not include a tool specifically for sentiment analysis. However, we do have [Project Name], which provides [related functionality]. Would that be of interest?"
+
+✅ Customer: "I need a blockchain-based security system. Can your team provide one?"
+✅ Agent: "Unfortunately, we do not have a blockchain-based security system in our current offerings. I'm happy to help with any other inquiries regarding our available solutions."
+
+Tone & Style:
+Maintain a professional, helpful, and customer-focused tone.
+Keep responses concise yet informative based on brochure content.
+If a solution exists, explain how it meets the customer's needs without overselling.
+If no solution exists, remain polite and transparent.
+
+First Brochure:
+{text1}
+
+Second Brochure:
+{text2}
+                    '''
+        
+        return genai.GenerativeModel("gemini-1.5-pro", system_instruction=final_prompt)
+    
+    def answer_question(self, question: str) -> str:
+        """Performs the API call to Gemini and returns the sentiment analysis response."""
+        response = self.model.generate_content(question)
+        return response.text
+
+if __name__ == "__main__":
+    analyzer = GeminiQanA()
+    response = analyzer.answer_question("Hello,how are you?")
+    print(response)