Update app.py

app.py

@@ -1,11 +1,4 @@
-import time
-import os
-import multiprocessing
-import torch
-import requests
-import asyncio
-import json
-import aiohttp
+import time, os, multiprocessing, torch, requests, asyncio, json, aiohttp
 from minivectordb.embedding_model import EmbeddingModel
 from minivectordb.vector_database import VectorDatabase
 from text_util_en_pt.cleaner import structurize_text, detect_language, Language
@@ -31,17 +24,17 @@ def index_and_search(query, text):
 
     # Indexing
     vector_db = VectorDatabase()
-    sentences = [s['sentence'] for s in structurize_text(text)]
+    sentences = [ s['sentence'] for s in structurize_text(text)]
 
     for idx, sentence in enumerate(sentences):
         sentence_embedding = model.extract_embeddings(sentence)
         vector_db.store_embedding(idx + 1, sentence_embedding, {'sentence': sentence})
-
+    
     embedding_time = time.time() - start
 
     # Retrieval
     start = time.time()
-    search_results = vector_db.find_most_similar(query_embedding, k=30)
+    search_results = vector_db.find_most_similar(query_embedding, k = 30)
     retrieval_time = time.time() - start
     return '\n'.join([s['sentence'] for s in search_results[2]]), embedding_time, retrieval_time
 
@@ -52,22 +45,18 @@ def generate_search_terms(message, lang):
         prompt = f"From the following text, generate some search terms: \"{message}\"\nYour answer should be just the most appropriate search term, and nothing else."
 
     url = "https://openrouter.ai/api/v1/chat/completions"
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {openrouter_key}"
-    }
-    body = {
-        "stream": False,
-        "models": [
-            "mistralai/mistral-7b-instruct:free",
-            "openchat/openchat-7b:free"
-        ],
-        "route": "fallback",
-        "max_tokens": 1024,
-        "messages": [
-            {"role": "user", "content": prompt}
-        ]
-    }
+    headers = { "Content-Type": "application/json",
+                "Authorization": f"Bearer {openrouter_key}" }
+    body = { "stream": False,
+             "models": [
+                    "mistralai/mistral-7b-instruct:free",
+                    "openchat/openchat-7b:free"
+             ],
+             "route": "fallback",
+             "max_tokens": 1024,
+             "messages": [
+                 {"role": "user", "content": prompt}
+             ] }
     
     response = requests.post(url, headers=headers, json=body)
     return response.json()['choices'][0]['message']['content']
@@ -131,22 +120,18 @@ async def predict(message, history):
     full_response += "\nResponse: "
     
     url = "https://openrouter.ai/api/v1/chat/completions"
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {openrouter_key}"
-    }
-    body = {
-        "stream": True,
-        "models": [
-            "mistralai/mistral-7b-instruct:free",
-            "openchat/openchat-7b:free"
-        ],
-        "route": "fallback",
-        "max_tokens": 1024,
-        "messages": [
-            {"role": "user", "content": prompt}
-        ]
-    }
+    headers = { "Content-Type": "application/json",
+                "Authorization": f"Bearer {openrouter_key}" }
+    body = { "stream": True,
+             "models": [
+                    "mistralai/mistral-7b-instruct:free",
+                    "openchat/openchat-7b:free"
+             ],
+             "route": "fallback",
+             "max_tokens": 1024,
+             "messages": [
+                 {"role": "user", "content": prompt}
+             ] }
 
     async with aiohttp.ClientSession() as session:
         async with session.post(url, headers=headers, json=body) as response:
@@ -173,26 +158,15 @@ async def predict(message, history):
                                 except Exception:
                                     pass
 
-# Define a custom theme with colors similar to the provided image
-custom_theme = gr.themes.Default(
-    primary_hue="purple",  # Custom primary color (similar to the purple in the image)
-    secondary_hue="indigo",  # Custom secondary color
-    font="Roboto",  # Custom font
-    rounded_corners=True,  # Enable rounded corners
-    background_color="#1a1a2e",  # Custom dark background color
-    text_color="#f0f0f0"  # Custom text color for better contrast
-)
-
-# Apply the custom theme to the Gradio interface
 gr.ChatInterface(
     predict,
-    title="Assistant Pro",
+    title="Live Web Chat",
+    description="",
     retry_btn=None,
     undo_btn=None,
     examples=[
         'What is the current sentiment of the Brazil election?',
         'Compare the current economies of China and India?',
-        'What are new shoe design trends in 2024?',
-    ],
-    theme=custom_theme  # Apply the custom theme
-).launch()
+        'What are new shoe design trends in 2024',
+    ]
+).launch()