Update app.py

app.py

@@ -891,65 +891,49 @@ from googlemaps import Client as GoogleMapsClient
 from diffusers import StableDiffusion3Pipeline
 import concurrent.futures
 from PIL import Image
-from flask import Flask, redirect, url_for, session
-from authlib.integrations.flask_client import OAuth
-
-# Initialize Flask app
-app = Flask(__name__)
-app.secret_key = os.urandom(24)
-
-# OAuth setup
-oauth = OAuth(app)
-google = oauth.register(
-    name='google',
-    client_id=os.environ['GOOGLE_CLIENT_ID'],
-    client_secret=os.environ['GOOGLE_CLIENT_SECRET'],
-    access_token_url='https://accounts.google.com/o/oauth2/token',
-    authorize_url='https://accounts.google.com/o/oauth2/auth',
-    authorize_params=None,
-    access_token_params=None,
-    refresh_token_url=None,
-    redirect_uri='http://localhost:7860/oauth2callback',
-    client_kwargs={'scope': 'openid profile email'},
-)
 
-@app.route('/')
-def home():
-    email = dict(session).get('email', None)
-    if email:
-        return f'Hello, {email}!'
-    return redirect('/login')
-
-@app.route('/login')
-def login():
-    return google.authorize_redirect(url_for('authorize', _external=True))
-
-@app.route('/logout')
-def logout():
-    session.pop('email', None)
-    return redirect('/')
-
-@app.route('/authorize')
-def authorize():
-    token = google.authorize_access_token()
-    resp = google.get('https://www.googleapis.com/oauth2/v1/userinfo')
-    user_info = resp.json()
-    session['email'] = user_info['email']
-    return redirect('/')
-
-@app.route('/oauth2callback')
-def oauth2callback():
-    token = google.authorize_access_token()
-    resp = google.get('https://www.googleapis.com/oauth2/v1/userinfo')
-    user_info = resp.json()
-    session['email'] = user_info['email']
-    return redirect('/')
-
-# Your Gradio interface and other functions go here
-
-# Initialize logging
+# Check if the token is already set in the environment variables
+hf_token = os.getenv("HF_TOKEN")
+
+if hf_token is None:
+    # If the token is not set, prompt for it (this should be done securely)
+    print("Please set your Hugging Face token in the environment variables.")
+else:
+    # Login using the token
+    login(token=hf_token)
+
+# Your application logic goes here
+print("Logged in successfully to Hugging Face Hub!")
+
+# Set up logging
 logging.basicConfig(level=logging.DEBUG)
 
+# Initialize OpenAI embeddings
+from langchain_openai import OpenAIEmbeddings, ChatOpenAI
+from langchain_pinecone import PineconeVectorStore
+from langchain.prompts import PromptTemplate
+from langchain.chains import RetrievalQA
+from langchain.chains.conversation.memory import ConversationBufferWindowMemory
+from langchain.agents import Tool, initialize_agent
+
+embeddings = OpenAIEmbeddings(api_key=os.environ['OPENAI_API_KEY'])
+
+# Initialize Pinecone
+from pinecone import Pinecone
+pc = Pinecone(api_key=os.environ['PINECONE_API_KEY'])
+
+index_name = "omaha-details"
+vectorstore = PineconeVectorStore(index_name=index_name, embedding=embeddings)
+retriever = vectorstore.as_retriever(search_kwargs={'k': 5})
+
+chat_model = ChatOpenAI(api_key=os.environ['OPENAI_API_KEY'], temperature=0, model='gpt-4o')
+
+conversational_memory = ConversationBufferWindowMemory(
+    memory_key='chat_history',
+    k=10,
+    return_messages=True
+)
+
 # Define function to get current time and date
 def get_current_time_and_date():
     now = datetime.now()
@@ -1170,32 +1154,6 @@ Keep the answer short and sweet and crisp. Always say "It was my pleasure!" at t
 Question: {question}
 Helpful Answer:"""
 
-# Initialize ChatOpenAI model
-from langchain_openai import OpenAIEmbeddings, ChatOpenAI
-from langchain_pinecone import PineconeVectorStore
-from langchain.prompts import PromptTemplate
-from langchain.chains import RetrievalQA
-from langchain.chains.conversation.memory import ConversationBufferWindowMemory
-from langchain.agents import Tool, initialize_agent
-
-embeddings = OpenAIEmbeddings(api_key=os.environ['OPENAI_API_KEY'])
-
-# Initialize Pinecone
-from pinecone import Pinecone
-pc = Pinecone(api_key=os.environ['PINECONE_API_KEY'])
-
-index_name = "omaha-details"
-vectorstore = PineconeVectorStore(index_name=index_name, embedding=embeddings)
-retriever = vectorstore.as_retriever(search_kwargs={'k': 5})
-
-chat_model = ChatOpenAI(api_key=os.environ['OPENAI_API_KEY'], temperature=0, model='gpt-4o')
-
-conversational_memory = ConversationBufferWindowMemory(
-    memory_key='chat_history',
-    k=10,
-    return_messages=True
-)
-
 QA_CHAIN_PROMPT_1 = PromptTemplate(input_variables=["context", "question"], template=template1)
 QA_CHAIN_PROMPT_2 = PromptTemplate(input_variables=["context", "question"], template=template2)
 
@@ -1297,25 +1255,7 @@ def generate_map(location_names):
             ).add_to(m)
     map_html = m._repr_html_()
     return map_html
-def transcribe_function(stream, new_chunk):
-    try:
-        sr, y = new_chunk[0], new_chunk[1]
-    except TypeError:
-        print(f"Error chunk structure: {type(new_chunk)}, content: {new_chunk}")
-        return stream, "", None
-
-    y = y.astype(np.float32) / np.max(np.abs(y))
-
-    if stream is not None:
-        stream = np.concatenate([stream, y])
-    else:
-        stream = y
 
-    result = pipe_asr({"array": stream, "sampling_rate": sr}, return_timestamps=False)
-
-    full_text = result.get("text", "")
-    
-    return stream, full_text, result
 def update_map_with_response(history):
     if not history:
         return ""
@@ -1385,6 +1325,10 @@ def update_images():
     image_3 = generate_image(hardcoded_prompt_3)
     return image_1, image_2, image_3
 
+def login_with_google():
+    response = requests.get('http://localhost:5000/login')
+    return response.url
+
 with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
     with gr.Row():
         with gr.Column():
@@ -1416,13 +1360,11 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])
 
     login_button = gr.Button("Login with Google")
-    login_button.click(fn=lambda: redirect('/login'))
+    login_button.click(fn=login_with_google)
 
 demo.queue()
 demo.launch(share=True)
 
-if __name__ == "__main__":
-    app.run(port=7860)