From 98f32c8086ec084d8c309816c1b3ff2cb904bc2a Mon Sep 17 00:00:00 2001
From: Codewithdark <144595403+codewithdark-git@users.noreply.github.com>
Date: Sat, 6 Apr 2024 13:11:35 +0500
Subject: [PATCH] Update app.py

---
 app.py | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index ee4ae9f34..dff47240d 100644
--- a/app.py
+++ b/app.py
@@ -1,4 +1,5 @@
 import streamlit as st
+import g4f
 from g4f.client import Client
 import sqlite3
 import google.generativeai as genai
@@ -53,8 +54,8 @@ def main():
         models = {
             "🚀 Airoboros 70B": "airoboros-70b",
             "🔮 Gemini Pro": "gemini-pro",
-            "📷 StabilityAI": "stabilityai/stable-diffusion-xl-base-1.0"
-
+            "📷 StabilityAI": "stabilityai/stable-diffusion-xl-base-1.0",
+            "🧨 GPT-3-turbo": g4f.models.default
         }
 
         columns = st.columns(3)  # Split the layout into three columns
@@ -130,6 +131,31 @@ def main():
                 generated_image = generate_image_from_model(prompt)
                 st.image(generated_image, caption="Generated Image", width=400)
 
+            elif selected_model == "GPT-3-turbo":
+                try:
+
+                    client = Client()
+                    response = client.chat.completions.create(
+                        messages=[{"role": "user", "content": user_input}],
+                        model=g4f.models.default,
+                    )
+
+                    # Extract the GPT response and print it
+                    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})
+
+                    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}")
 
             else:
                 try: