Update app.py

app.py

@@ -5,14 +5,8 @@ import sqlite3
 import google.generativeai as genai
 # import pyttsx3
 import pyperclip
-import requests
-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_huggingface_Api_key"}
-
 def local_css(file_name):
     with open(file_name) as f:
         st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
@@ -32,25 +26,6 @@ try:
 except Exception as e:
     st.error(f"An error occurred: {e}")
 
-def generate_image_from_model(prompt):
-    try:
-        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-        response.raise_for_status()  # Raise an error for bad responses
-        image_bytes = response.content
-        if not image_bytes:
-            raise ValueError("Empty image content received from the API")
-        image = Image.open(io.BytesIO(image_bytes))
-        return image
-    except Exception as e:
-        st.error(f"Error generating image from model: {e}")
-        
-        return None
-
-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():
@@ -63,8 +38,11 @@ def main():
 
         models = {
             "🚀 Airoboros 70B": "airoboros-70b",
+            "👑 Gemini 1.0": "gemini-1.0-pro",
+            "🧨 Gemini 1.0 Pro ": "gemini-1.0-pro-001",
+            "⚡ Gemini 1.0 pro latest": "gemini-1.0-pro-latest",
             "🔮 Gemini Pro": "gemini-pro",
-            "📷 StabilityAI": "stabilityai/stable-diffusion-xl-base-1.0"
+            "🎉 Gemini pro vision": "gemini-pro-vision"
         }
 
         columns = st.columns(3)  # Split the layout into three columns
@@ -106,40 +84,41 @@ def main():
         user_input = st.chat_input("Ask Anything ...")
 
         if user_input:
-            if selected_model == "gemini-pro":
+            if selected_model == "airoboros-70b":
                 try:
+                    client = Client()
+                    response = client.chat.completions.create(
+                        model=models[selected_model_display_name],
+                        messages=[{"role": "user", "content": user_input}],
+                    )
+                    bot_response = response.choices[0].message.content
 
-                    if user_input.startswith("/image"):
-                        prompt = user_input[len("/image"):].strip()  # Extract prompt after "/image"
+                    st.session_state.chat_history.append({"role": "user", "content": user_input})
+                    st.session_state.chat_history.append({"role": "bot", "content": bot_response})
 
-                        # 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
+                    # 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()
 
-                        # Generate image based on the generated text prompt
-                        generated_image = generate_image(bot_response)
+                    # Display chat history
+                    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"])
 
-                        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
-                        if generated_image is not None:
-                            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:
-                            st.error("Failed to generate image. Check logs for details.")
-
-                    else:
-                        GOOGLE_API_KEY = "your_gemini_Api_key"
+
+                except Exception as e:
+                    st.error(f"An error occurred: {e}")
+
+            else:
+                try:
+                        GOOGLE_API_KEY = "your_Gemini_Api_key"
                         genai.configure(api_key=GOOGLE_API_KEY)
-                        model = genai.GenerativeModel('gemini-1.0-pro')
+                        model = genai.GenerativeModel(selected_model)
                         prompt = user_input
                         response = model.generate_content(prompt)
                         bot_response = response.candidates[0].content.parts[0].text
@@ -163,48 +142,7 @@ def main():
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
 
-            elif selected_model == "stabilityai/stable-diffusion-xl-base-1.0":
-                prompt = user_input
-                generated_image = generate_image_from_model(prompt)
-                if generated_image is not None:
-                    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:
-                    st.error("Failed to generate image. Check logs for details.")
 
-            else:
-                try:
-                    client = Client()
-                    response = client.chat.completions.create(
-                        model=models[selected_model_display_name],
-                        messages=[{"role": "user", "content": user_input}],
-                    )
-                    bot_response = response.choices[0].message.content
-
-                    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()
-
-                    # Display chat history
-                    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}")