Update app.py

app.py

@@ -1,121 +1,91 @@
-import streamlit as st
-import json
+import sys
+from PyQt5.QtWidgets import QApplication, QWidget, QVBoxLayout, QLabel, QTextEdit, QPushButton, QLineEdit, QCheckBox, QHBoxLayout
+from PyQt5.QtGui import QFont, QColor, QPalette
+from PyQt5.QtCore import Qt
 from huggingface_hub import InferenceClient
-import streamlit.web.bootstrap
-streamlit.web.bootstrap.run('app.py', '', [], {})
 
-# Set page config
-st.set_page_config(page_title="VarunGPT", page_icon="🤖", layout="centered")
-
-# **Custom CSS for Styling**
-st.markdown(
-    """
-    <style>
-        body {
-            background: linear-gradient(to right, #00c6ff, #0072ff);
-            color: white;
-        }
-        .stButton>button {
-            background-color: #0044cc;
-            color: white;
-            border-radius: 10px;
-            transition: 0.3s;
-            font-size: 18px;
-            padding: 10px 20px;
-            border: none;
-        }
-        .stButton>button:hover {
-            background-color: #002266;
-            transform: scale(1.1);
-            box-shadow: 0px 0px 15px rgba(0, 255, 255, 0.6);
-        }
-        .chat-bubble-user {
-            background-color: #0072ff;
-            padding: 10px;
-            border-radius: 15px;
-            margin: 5px;
-            text-align: right;
-            color: white;
-            display: inline-block;
-            max-width: 100%;
-        }
-        .chat-bubble-bot {
-            background-color: #0044cc;
-            padding: 10px;
-            border-radius: 15px;
-            margin: 5px;
-            text-align: left;
-            color: white;
-            display: inline-block;
-            max-width: 100%;
-        }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
-# **App Title**
-st.title("🤖 VarunGPT")
-st.write("Made in India 🇮🇳")
-st.write("Chat and get programming solutions, news, and much more! 😊")
-
-# **API Key Input for Login**
-st.subheader("🔑 Enter your Hugging Face API Key")
-api_key = st.text_input("API Key", type="password")
-
-# **Validate API Key**
-if api_key:
-    st.success("API Key entered successfully!")
-    client = InferenceClient(api_key=api_key)
-else:
-    st.warning("Please enter a valid API key to continue.")
-    st.stop()
-
-# **Sidebar Category**
-theme = st.sidebar.selectbox("Category", ["Fine-Tuned"])
-
-# **Initialize Chat History**
-if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []
-
-# **User Input Box**
-user_input = st.text_input("You: ", "")
-
-# **Define function to query Hugging Face Inference API**
-def query_huggingface(prompt):
-    messages = [{"role": "user", "content": prompt}]
+class VarunGPTApp(QWidget):
+    def __init__(self):
+        super().__init__()
+        self.initUI()
+        self.chat_history = ""
     
-    try:
-        # Make the API call to get the model's response
-        completion = client.chat.completions.create(
-            model="mistralai/Mistral-7B-Instruct-v0.3", 
-            messages=messages, 
-            max_tokens=500
-        )
+    def initUI(self):
+        self.setWindowTitle("VarunGPT - AI Chatbot")
+        self.setGeometry(200, 200, 700, 550)
         
-        # Extract and return the generated content from the response
-        return completion["choices"][0]["message"]["content"]
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# **Handle user input and generate AI response**
-if st.button("Send"):
-    if user_input:
-        # Store user message
-        st.session_state.chat_history.append(("You", user_input))
-
-        # Query Hugging Face API (Mistral-7B model)
-        ai_response = query_huggingface(user_input)
-
-        # Store AI response
-        st.session_state.chat_history.append(("VarunGPT", ai_response))
-
-        # Show balloon animation when a message is sent
-        st.balloons()
+        palette = self.palette()
+        palette.setColor(QPalette.Window, QColor(20, 20, 20))
+        self.setPalette(palette)
+        
+        layout = QVBoxLayout()
+        
+        self.title = QLabel("VarunGPT", self)
+        self.title.setFont(QFont("Arial", 22, QFont.Bold))
+        self.title.setStyleSheet("color: #ff9800; text-align: center;")
+        self.title.setAlignment(Qt.AlignCenter)
+        layout.addWidget(self.title)
+        
+        self.api_key_input = QLineEdit(self)
+        self.api_key_input.setPlaceholderText("Enter API Key")
+        self.api_key_input.setEchoMode(QLineEdit.Password)
+        self.api_key_input.setStyleSheet("background-color: #222; color: white; padding: 8px; border-radius: 5px;")
+        layout.addWidget(self.api_key_input)
+        
+        self.theme_checkbox = QCheckBox("Enable Advanced Reasoning")
+        self.theme_checkbox.setStyleSheet("color: white;")
+        layout.addWidget(self.theme_checkbox)
+        
+        self.user_input = QLineEdit(self)
+        self.user_input.setPlaceholderText("Ask anything...")
+        self.user_input.setStyleSheet("background-color: #222; color: white; padding: 8px; border-radius: 5px;")
+        layout.addWidget(self.user_input)
+        
+        self.send_button = QPushButton("Send", self)
+        self.send_button.setStyleSheet("background-color: #ff9800; color: white; padding: 10px; border-radius: 5px;")
+        self.send_button.clicked.connect(self.handleUserInput)
+        layout.addWidget(self.send_button)
+        
+        self.chat_display = QTextEdit(self)
+        self.chat_display.setReadOnly(True)
+        self.chat_display.setStyleSheet("background-color: #111; color: white; padding: 10px; border-radius: 5px;")
+        layout.addWidget(self.chat_display)
+        
+        self.setLayout(layout)
+    
+    def handleUserInput(self):
+        api_key = self.api_key_input.text().strip()
+        if not api_key:
+            self.chat_display.setText("Error: API Key is required.")
+            return
+        
+        client = InferenceClient(api_key=api_key)
+        
+        user_text = self.user_input.text()
+        if user_text:
+            self.chat_history += f"You: {user_text}\n"
+            self.chat_display.setText(self.chat_history)
+            
+            ai_response = self.query_huggingface(client, user_text)
+            self.chat_history += f"VarunGPT: {ai_response}\n"
+            self.chat_display.setText(self.chat_history)
+            
+            self.user_input.clear()
+    
+    def query_huggingface(self, client, prompt):
+        messages = [{"role": "user", "content": prompt}]
+        try:
+            completion = client.chat.completions.create(
+                model="mistralai/Mistral-7B-Instruct-v0.3", 
+                messages=messages, 
+                max_tokens=500
+            )
+            return completion["choices"][0]["message"]["content"]
+        except Exception as e:
+            return f"Error: {str(e)}"
 
-# **Display chat history with bubble effects**
-for speaker, message in st.session_state.chat_history:
-    if speaker == "You":
-        st.markdown(f"<div class='chat-bubble-user'><b>🧑‍💻 {speaker}:</b> {message}</div>", unsafe_allow_html=True)
-    else:
-        st.markdown(f"<div class='chat-bubble-bot'><b>🤖 {speaker}:</b> {message}</div>", unsafe_allow_html=True)
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    window = VarunGPTApp()
+    window.show()
+    sys.exit(app.exec_())