Update app.py

app.py

@@ -1,16 +1,17 @@
 import streamlit as st
+import g4f
 from g4f.client import Client
 import sqlite3
 import google.generativeai as genai
 # import pyttsx3
-# import pyperclip
+import pyperclip
 import requests
-import cv2
-import numpy as np
+from PIL import Image
+import io
 
 
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-headers = {"Authorization": "Bearer Your_hugging_face_Api_key"}
+headers = {"Authorization": "Bearer Your_huggingface_Api_key"}
 
 def local_css(file_name):
     with open(file_name) as f:
@@ -34,12 +35,14 @@ except Exception as e:
 def generate_image_from_model(prompt):
     response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     image_bytes = response.content
-    # Convert image bytes to a NumPy array
-    nparr = np.frombuffer(image_bytes, np.uint8)
-    # Decode the image array
-    image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+    image = Image.open(io.BytesIO(image_bytes))
     return image
 
+def generate_image(prompt):
+    response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+    image_bytes = response.content
+    image = Image.open(io.BytesIO(image_bytes))
+    return image
 
 # Streamlit app
 def main():
@@ -54,7 +57,6 @@ def main():
             "🚀 Airoboros 70B": "airoboros-70b",
             "🔮 Gemini Pro": "gemini-pro",
             "📷 StabilityAI": "stabilityai/stable-diffusion-xl-base-1.0"
-
         }
 
         columns = st.columns(3)  # Split the layout into three columns
@@ -98,29 +100,54 @@ def main():
         if user_input:
             if selected_model == "gemini-pro":
                 try:
-                    GOOGLE_API_KEY = "your_Gemini_Api_key"
-                    genai.configure(api_key=GOOGLE_API_KEY)
-                    model = genai.GenerativeModel('gemini-pro')
-                    prompt = user_input
-                    response = model.generate_content(prompt)
-                    bot_response = response.candidates[0].content.parts[0].text
-
-                    st.session_state.chat_history.append({"role": "user", "content": user_input})
-                    st.session_state.chat_history.append({"role": "bot", "content": bot_response})
-
-                    # Store chat in the database
-                    for chat in st.session_state.chat_history:
-                        c.execute("INSERT INTO chat_history VALUES (?, ?, ?)",
-                                  (st.session_state.conversation_id, chat["role"], chat["content"]))
-                    conn.commit()
-
-                    for index, chat in enumerate(st.session_state.chat_history):
-                        with st.chat_message(chat["role"]):
-                            if chat["role"] == "user":
-                                st.markdown(chat["content"])
-                            elif chat["role"] == "bot":
-                                st.markdown(chat["content"])
 
+                    if user_input.startswith("/image"):
+                        prompt = user_input[len("/image"):].strip()  # Extract prompt after "/image"
+
+                        # Use Gemini Pro to generate content based on the prompt
+                        GOOGLE_API_KEY = "AIzaSyC8_gwU5LSVQJk3iIXyj5xJ94ArNK11dXU"
+                        genai.configure(api_key=GOOGLE_API_KEY)
+                        model = genai.GenerativeModel('gemini-1.0-pro')
+                        response = model.generate_content(prompt)
+                        bot_response = response.candidates[0].content.parts[0].text
+
+                        # Generate image based on the generated text prompt
+                        generated_image = generate_image(bot_response)
+
+                        st.session_state.chat_history.append({"role": "user", "content": user_input})
+                        st.session_state.chat_history.append({"role": "bot", "content": generated_image})
+
+                        # Display the generated image
+                        for index, chat in enumerate(st.session_state.chat_history):
+                            with st.chat_message(chat["role"]):
+                                if chat["role"] == "user":
+                                    st.markdown(user_input)
+                                elif chat["role"] == "bot":
+                                    st.image(generated_image, width=400)
+
+                    else:
+                        GOOGLE_API_KEY = "your_Gemini_Api_key"
+                        genai.configure(api_key=GOOGLE_API_KEY)
+                        model = genai.GenerativeModel('gemini-1.0-pro')
+                        prompt = user_input
+                        response = model.generate_content(prompt)
+                        bot_response = response.candidates[0].content.parts[0].text
+
+                        st.session_state.chat_history.append({"role": "user", "content": user_input})
+                        st.session_state.chat_history.append({"role": "bot", "content": bot_response})
+
+                        # Store chat in the database
+                        for chat in st.session_state.chat_history:
+                            c.execute("INSERT INTO chat_history VALUES (?, ?, ?)",
+                                      (st.session_state.conversation_id, chat["role"], chat["content"]))
+                        conn.commit()
+
+                        for index, chat in enumerate(st.session_state.chat_history):
+                            with st.chat_message(chat["role"]):
+                                if chat["role"] == "user":
+                                    st.markdown(chat["content"])
+                                elif chat["role"] == "bot":
+                                    st.markdown(chat["content"])
 
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
@@ -128,10 +155,12 @@ def main():
             elif selected_model == "stabilityai/stable-diffusion-xl-base-1.0":
                 prompt = user_input
                 generated_image = generate_image_from_model(prompt)
-                # Display the image using OpenCV
-                cv2.imshow("Generated Image", generated_image)
-                cv2.waitKey(0)
-                cv2.destroyAllWindows()
+                for index, chat in enumerate(st.session_state.chat_history):
+                    with st.chat_message(chat["role"]):
+                        if chat["role"] == "user":
+                            st.markdown(user_input)
+                        elif chat["role"] == "bot":
+                            st.image(generated_image, width=400)
 
             else:
                 try: