Update app.py

app.py

@@ -10,16 +10,44 @@ import requests
 from bs4 import BeautifulSoup
 import cv2
 from io import BytesIO
+import torch
 
 login(token=os.getenv("chatbot"))
 generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
 bg_to_en = pipeline("translation", model="Helsinki-NLP/opus-mt-bg-en")
 en_to_bg = pipeline("translation", model="Helsinki-NLP/opus-mt-en-bg")
+
 # Load BLIP for image captioning
 blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 blip_model.to("cuda" if torch.cuda.is_available() else "cpu")
 
+def describe_image(image):
+    inputs = blip_processor(images=image, return_tensors="pt").to(blip_model.device)
+    out = blip_model.generate(**inputs)
+    caption = blip_processor.decode(out[0], skip_special_tokens=True)
+    return caption
+
+def describe_video(video_path):
+    cap = cv2.VideoCapture(video_path)
+    success, frame = cap.read()
+    cap.release()
+    if success:
+        img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+        return describe_image(img)
+    else:
+        return "Could not read video frame."
+
+def extract_text_from_url(url):
+    try:
+        response = requests.get(url, timeout=5)
+        soup = BeautifulSoup(response.text, 'html.parser')
+        paragraphs = soup.find_all('p')
+        text = ' '.join([p.text for p in paragraphs])
+        return text[:2000]  # keep it short
+    except Exception as e:
+        return f"Failed to fetch URL: {str(e)}"
+
 def generate_response(user_input, top_p, temperature, chat_counter, chatbot, history, image=None, video=None, url=None, request: gr.Request = None):
     lang = detect(user_input)
     print(f"Detected language: {lang}")
@@ -27,9 +55,11 @@ def generate_response(user_input, top_p, temperature, chat_counter, chatbot, his
     # Translate if needed
     if lang == "bg":
         user_input_translated = bg_to_en(user_input)[0]["translation_text"]
+    else:
+        user_input_translated = user_input
+
+    prompt = ""
 
-            prompt = ""
-        
     # Multimodal additions
     if image is not None:
         try:
@@ -52,68 +82,6 @@ def generate_response(user_input, top_p, temperature, chat_counter, chatbot, his
     # Append user message to prompt
     prompt += f"User said:\n{user_input_translated}\n\nGive simple, friendly, clear advice:\n"
 
-    else:
-        user_input_translated = user_input
-
-def describe_image(image):
-    inputs = blip_processor(images=image, return_tensors="pt").to(blip_model.device)
-    out = blip_model.generate(**inputs)
-    caption = blip_processor.decode(out[0], skip_special_tokens=True)
-    return caption
-
-def describe_video(video_path):
-    cap = cv2.VideoCapture(video_path)
-    success, frame = cap.read()
-    cap.release()
-    if success:
-        img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-        return describe_image(img)
-    else:
-        return "Could not read video frame."
-
-def extract_text_from_url(url):
-    try:
-        response = requests.get(url, timeout=5)
-        soup = BeautifulSoup(response.text, 'html.parser')
-        paragraphs = soup.find_all('p')
-        text = ' '.join([p.text for p in paragraphs])
-        return text[:2000]  # keep it short
-    except Exception as e:
-        return f"Failed to fetch URL: {str(e)}"
-
-    # Create prompt
-    prompt = f"""
-Comprehensive E-commerce Bot Prompt:
-Role and Persona:
-You are an exceptionally helpful, friendly, and clear e-commerce business assistant, specifically designed for beginners and aspiring entrepreneurs. Your primary goal is to guide users through the process of establishing, developing, and scaling their online businesses, drawing upon the proven strategies and real-world experience embodied by the Masterbrand e-commerce training program. You should embody the spirit of a seasoned mentor who simplifies complex e-commerce concepts into actionable, easy-to-understand advice.
-Core Functionality and Expertise:
-Your expertise is rooted in the principles taught by Masterbrand, which focuses on building successful e-commerce brands from the ground up. You are proficient in:
-1. Foundational E-commerce Setup: Guiding users on how to start an e-commerce brand from scratch, including initial ideation, niche selection, and basic platform setup.
-2. Business Development and Growth: Providing insights into developing online businesses, implementing effective growth strategies, and structuring a business for long-term success.
-3. Strategic Implementation: Offering practical advice on applying working strategies and business structures that lead to real sales and tangible results.
-4. Brand Building: Assisting users in transforming their initial ideas into established, profitable brands.
-5. Market Context (Bulgaria): While your advice is universally applicable, you understand the nuances of the e-commerce landscape, particularly in regions like Bulgaria, given Masterbrand\'s origin as the #1 ECOM training in Bulgaria.
-6. Learning Resource Guidance: You can refer to the types of resources found in Masterbrand (e.g., video lessons, modules, practical resources) to explain concepts or suggest learning paths.
-Tone and Communication Style:
-Your communication should always be:
-- Simple and Clear: Avoid jargon where possible, and explain complex terms in an accessible manner.
-- Friendly and Encouraging: Maintain a supportive and motivating tone, especially for beginners who may feel overwhelmed.
-- Action-Oriented: Provide advice that is practical and can be immediately applied by the user.
-- Empathetic: Acknowledge the challenges beginners face and offer reassurance, similar to how Atanas Peltekov, the founder of Masterbrand, emphasizes overcoming initial hurdles.
-Constraints and Limitations:
-You are an assistant, not a decision-maker. Your role is to provide information and guidance, not to make business decisions for the user.
-You should not provide financial or legal advice. Always recommend consulting with professionals for such matters.
-Your knowledge is based on the principles of successful e-commerce as observed and taught by Masterbrand. While comprehensive, it is not exhaustive of all possible e-commerce strategies.
-Example Interaction Flow:
-User: "I want to start an online store, but I have no idea where to begin."
-Bot: "That\'s a fantastic goal! Many successful e-commerce brands started exactly where you are now. To begin, let\'s think about what you\'re passionate about or what problem you want to solve for customers. This will help us choose a niche. Once we have a niche, we can explore simple ways to set up your first online presence. What kind of products or services are you considering?"
-Integration with User Input:
-Always consider the user\'s specific input and tailor your advice accordingly. If the user provides a specific question or scenario, address it directly while integrating the core principles of e-commerce success.
-User said:
-{user_input_translated}
-Give simple, friendly, clear advice:
-"""
-
     # Generate response
     response_text = generator(prompt, max_length=200, top_p=top_p, temperature=temperature, do_sample=True)[0]["generated_text"]
 
@@ -125,10 +93,12 @@ Give simple, friendly, clear advice:
     history.append({"role": "assistant", "content": response_text})
 
     chat_counter += 1
-    return history, history, chat_counter, "✅ Success", gr.update(value='\\'), gr.update(interactive=True)
+    return history, history, chat_counter, "✅ Success", gr.update(value=\'\
+'), gr.update(interactive=True)
 
 def reset_textbox():
-    return gr.update(value='', interactive=False), gr.update(interactive=False)
+    return gr.update(value=\'\
+'), gr.update(interactive=False)
 
 # ==== Enhanced Custom CSS with Masterbrand Styling ====
 custom_css = f""" 
@@ -191,7 +161,7 @@ custom_css = f"""
     background: rgba(26, 26, 26, 0.95) !important;
     border: 2px solid #ff0000 !important; /* Changed to red */
     border-radius: 25px !important;
-    box-shadow: 0 15px 40px rgba(255, 0, 0, 0.2), inset 0 1px 0 rgba(255, 255, 255, 0.1) !important;
+    box-shadow: 0 15px 40px rgba(255, 0, 0, 0.2), inset 0 1px 0 rgba(255, 255, 255, 0.1) !important; /* Changed to red */
     margin-bottom: 2rem !important;
     backdrop-filter: blur(10px) !important;
 }}
@@ -209,7 +179,7 @@ custom_css = f"""
 }}
 
 .input-container textarea {{
-    background: linear-gradient(135deg, #1a1a1a 0%, #2a2a2a 100%) !important;
+    background: linear-gradient(135deg, #1a1a1a 0%, #2a2a1a 100%) !important;
     border: 2px solid #ff0000 !important; /* Changed to red */
     border-radius: 15px !important;
     color: #ffffff !important;
@@ -287,7 +257,7 @@ custom_css = f"""
 
 /* Enhanced chat styling */
 .message.user {{
-    background: linear-gradient(135deg, #2a2a2a 0%, #3a3a3a 100%) !important;
+    background: linear-gradient(135deg, #2a2a2a 0%, #3a3a2a 100%) !important;
     border-left: 4px solid #ff0000 !important; /* Changed to red */
     margin-left: 2rem !important;
 }}
@@ -362,15 +332,17 @@ with gr.Blocks(theme=masterbrand_theme, css=custom_css, title="MasterBrand AI As
 
             inputs = gr.Textbox(
                 placeholder="Ask me anything about your e-commerce business... 💡",
-                image_input = gr.Image(label="Upload an Image (optional)", type="pil", optional=True)
-                video_input = gr.Video(label="Upload a Video (optional)", optional=True)
-                url_input = gr.Textbox(label="Paste a URL (optional)", placeholder="https://example.com", optional=True)
                 label="Your Question",
                 lines=3,
                 elem_classes=["input-container"],
                 show_label=True
             )
 
+            image_input = gr.Image(label="Upload an Image (optional)", type="pil", optional=True)
+            video_input = gr.Video(label="Upload a Video (optional)", optional=True)
+            url_input = gr.Textbox(label="Paste a URL (optional)", placeholder="https://example.com", optional=True)
+
+
             with gr.Row():
                 submit_btn = gr.Button(
                     "🚀 Get Expert Advice",
@@ -423,18 +395,18 @@ with gr.Blocks(theme=masterbrand_theme, css=custom_css, title="MasterBrand AI As
 
     # Event Handlers
     inputs.submit(reset_textbox, [], [inputs, submit_btn], queue=False)
-inputs.submit(
-    generate_response,
-    [inputs, top_p, temperature, chat_counter, chatbot, state, image_input, video_input, url_input],
-    [chatbot, state, chat_counter, status_box, inputs, submit_btn]
-)
+    inputs.submit(
+        generate_response,
+        [inputs, top_p, temperature, chat_counter, chatbot, state, image_input, video_input, url_input],
+        [chatbot, state, chat_counter, status_box, inputs, submit_btn]
+    )
 
-submit_btn.click(reset_textbox, [], [inputs, submit_btn], queue=False)
-submit_btn.click(
-    generate_response,
-    [inputs, top_p, temperature, chat_counter, chatbot, state, image_input, video_input, url_input],
-    [chatbot, state, chat_counter, status_box, inputs, submit_btn]
-)
+    submit_btn.click(reset_textbox, [], [inputs, submit_btn], queue=False)
+    submit_btn.click(
+        generate_response,
+        [inputs, top_p, temperature, chat_counter, chatbot, state, image_input, video_input, url_input],
+        [chatbot, state, chat_counter, status_box, inputs, submit_btn]
+    )
 
 
 # ==== Launch ====
@@ -442,3 +414,4 @@ if __name__ == "__main__":
     demo.queue(max_size=10).launch(server_name="0.0.0.0", server_port=7860)
 
 
+