prompt changes

app.py

@@ -2,101 +2,156 @@ import streamlit as st
 import os
 import re
 from query_chat import GeminiQanA
-from docx import Document
 
 def extract_text_from_txt(file_path):
-    with open(file_path, "r", encoding="utf-8") as file:
-        return " ".join([line.strip() for line in file.readlines() if line.strip()])
+    # In a real scenario, handle FileNotFoundError
+    try:
+        with open(file_path, "r", encoding="utf-8") as file:
+            return " ".join([line.strip() for line in file.readlines() if line.strip()])
+    except FileNotFoundError:
+        st.warning(f"File not found: {file_path}. Using placeholder text.")
+        return f"Placeholder text for {os.path.basename(file_path)}"
+
 
 @st.cache_resource()
 def load_chatbot():
+    print("Attempting to load chatbot...") # Add print statement
     with st.spinner("Loading project information..."):
+        # Use dummy paths if Files/ directory doesn't exist or is empty
         doc1_text = extract_text_from_txt("Files/brochure_1.txt")
         doc2_text = extract_text_from_txt("Files/brochure_2.txt")
-    
-    return GeminiQanA(doc1_text, doc2_text)
+        chatbot_instance = GeminiQanA(doc1_text, doc2_text)
+        print("Chatbot loaded.") # Add print statement
+        return chatbot_instance
 
 # 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)
+# Use a placeholder if the logo doesn't exist
+logo_path = "zega_logo.PNG"
+if os.path.exists(logo_path):
+    st.sidebar.image(logo_path, width=300)
+else:
+    st.sidebar.warning("Logo zega_logo.PNG not found.")
 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
+# Load API key safely (using dummy key for this example if not set)
 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.")
-
-# Initialize chatbot
-chatbot = load_chatbot()
-
-# Initialize chat session
+    # For local testing without secrets, you might uncomment the next line
+    # os.environ["GOOGLE_API_KEY"] = "YOUR_DUMMY_OR_REAL_API_KEY"
+    st.sidebar.warning("Google API key not found in Streamlit secrets. Using demo mode.")
+
+# --- Initialize Chatbot ---
+# Ensure chatbot is loaded before defining callbacks that use it
+try:
+    chatbot = load_chatbot()
+except Exception as e:
+    st.error(f"Failed to load chatbot: {e}")
+    st.stop() # Stop execution if chatbot fails to load
+
+# --- Initialize Session State ---
 if "messages" not in st.session_state:
     st.session_state.messages = []
+# Initialize the input field state (important!)
+if "question_input" not in st.session_state:
+    st.session_state.question_input = ""
+
+
+# --- Callback Functions ---
+def handle_ask_ai():
+    """Callback function for the 'Ask AI' button."""
+    question = st.session_state.question_input # Get question from state
+    if question: # Proceed only if there is a question
+        # Append user question
+        st.session_state.messages.append({"role": "user", "content": question})
+
+        # Generate AI response
+        # Use a spinner *outside* the callback if needed, or understand
+        # the UI might briefly freeze if generation is long.
+        # For simplicity here, we generate directly.
+        try:
+            answer = chatbot.answer_question(question)
+            # Append AI response
+            st.session_state.messages.append({"role": "assistant", "content": answer})
+        except Exception as e:
+            st.error(f"Error getting answer from chatbot: {e}")
+            # Optionally remove the user message if the AI failed
+            # st.session_state.messages.pop()
+
+        # ***** Clear the input state variable AFTER processing *****
+        st.session_state.question_input = ""
+    # No explicit st.rerun() needed here, on_click handles the rerun
+
+
+def handle_clear_chat():
+    """Callback function for the 'Clear Chat' button."""
+    st.session_state.messages = []
+    chatbot.clear_conv_history()
+    st.session_state.question_input = "" # Clear the input field state as well
+    # No explicit st.rerun() needed here, on_click handles the rerun
 
-# Chat UI
+# --- Chat UI ---
 st.title("📄 Zega AI Sales Agent")
 
 # Function to display messages and replace image tags with actual images
 def display_message(role, content):
-    with st.chat_message(role):
-        # Find all image tags in the response
-        image_tags = re.findall(r"\[(.*?\.png)]", content)
-        
+     with st.chat_message(role):
+        # Find all image tags in the response (Corrected regex)
+        image_tags = re.findall(r"\[(.*?\.png)\]", content) # Find content inside brackets ending with .png
+
         # Split response by image tags and process separately
-        parts = re.split(r"\[(.*?\.png)]", content)
-        
+        parts = re.split(r"(\[.*?\.png\])", content) # Keep the delimiters
+
         for part in parts:
-            if part in image_tags:  
-                # If it's an image tag, check if the image exists and display it
-                image_path = f"ZegaPos/{part}"
+            if not part: # Skip empty strings from split
+                continue
+            match = re.match(r"\[(.*?\.png)\]", part) # Check if the part IS an image tag
+            if match:
+                image_filename = match.group(1)
+                # Assuming images are in a subfolder relative to the script
+                # Make sure the ZegaPos folder exists or handle the error
+                image_folder = "ZegaPos"
+                if not os.path.isdir(image_folder):
+                     st.warning(f"Image folder '{image_folder}' not found.")
+                     st.markdown(f"_{image_filename}_") # Display filename as text fallback
+                     continue # Skip trying to display the image
+
+                image_path = os.path.join(image_folder, image_filename)
                 if os.path.exists(image_path):
                     st.image(image_path, use_container_width=True)
                 else:
-                    st.markdown(f"⚠️ Image `{part}` not found.")
+                    st.markdown(f"⚠️ Image `{image_filename}` not found at `{image_path}`.")
             else:
                 # Otherwise, display text
                 st.markdown(part)
 
-# Display chat history
+
+# --- Display Chat History ---
+# This loop runs on every rerun, displaying the current state of messages
 for message in st.session_state.messages:
     display_message(message["role"], message["content"])
 
-# User Input
-question = st.text_area("Ask a question about Zega AI:", height=100)
+# --- User Input ---
+# This widget's value is controlled by st.session_state.question_input
+st.text_area(
+    "Ask a question about Zega AI:",
+    height=100,
+    key="question_input" # Links to st.session_state.question_input
+)
 
-# Chat Actions
+# --- Chat Actions ---
 col1, col2 = st.columns([3, 1])
 with col1:
-    ask_button = st.button("💬 Ask AI")
+    # Attach the callback function to the button's on_click parameter
+    st.button("💬 Ask AI", on_click=handle_ask_ai)
 with col2:
-    clear_button = st.button("🗑️ Clear Chat")
+    # Attach the callback function to the button's on_click parameter
+    st.button("🗑️ Clear Chat", on_click=handle_clear_chat)
 
-# Clear chat history
-if clear_button:
-    st.session_state.messages = []
-    chatbot.clear_conv_history()
-    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
-    display_message("user", 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 image handling
-    display_message("assistant", answer)
-        
-    st.rerun()
+# --- No processing logic needed here anymore ---
+# The callbacks handle the logic when the buttons are clicked.
+# Streamlit automatically reruns after a callback, updating the UI.

query_chat.py

@@ -3,7 +3,8 @@ import os
 
 class GeminiQanA:
     def __init__(self, text1: str = '', text2: str = ''):
-        """Initializes the Gemini question-answering model with brochures and conversation history."""
+        """Initializes the Gemini question-answering model with texts
+     and conversation history."""
         self.api_key = os.getenv("GOOGLE_API_KEY")
         genai.configure(api_key=self.api_key)
         self.text1 = text1
@@ -14,36 +15,37 @@ class GeminiQanA:
     def _load_model(self):
         """Loads the generative AI model without the conversation history (history will be passed dynamically)."""
         system_instruction = 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.
+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 texts that detail the available projects and their features. Your goal is to provide accurate and honest responses based solely on the information within these texts.
 
 ---
 
 ## Guidelines for Responses:
 
 ### 1. Accuracy & Honesty
-- Only provide responses based on the brochures.
+- Only provide responses based on the texts.
 - 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.
+- If information is not available in the texts, do not speculate—politely inform the customer that the requested details are not available.
+- Avoid saying that the answers are taken from a text and pretend you know them by heart
 
 ### 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.
+- When a customer asks about what our team can do, provide information only from the texts.
+- If asked about past projects, refer only to those explicitly mentioned in the texts.
+- If the customer asks for additional details not found in the texts, 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 customer presents a problem, check if a project in the texts 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. Handling Image Tags in Responses
-- Some solutions in the brochures may be followed by an image tag in the format `[solution_name.png]`.
+- Some solutions in the texts may be followed by an image tag in the format `[solution_name.png]`.
 - If an image tag appears in the provided brochure text, **include it exactly as it is when presenting the solution**.
-- Do not generate new image tags—only use them when they appear in the brochures.
+- Do not generate new image tags—only use them when they appear in the texts.
 
 ### 5. 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 make guarantees beyond what is stated in the texts.
 - Do not offer speculative solutions that are not explicitly supported by the documents.
 
 ---
@@ -56,12 +58,12 @@ You are a sales agent responsible for assisting customers by answering questions
 
 ---
 
-## Brochure Content:
+## Text Content:
 
-### First Brochure:
+### First text:
 {self.text1}
 
-### Second Brochure:
+### Second text:
 {self.text2}
 '''
         return genai.GenerativeModel("gemini-2.0-flash", system_instruction=system_instruction)