Update app.py

app.py

@@ -2,51 +2,55 @@ import streamlit as st
 import os
 from openai import OpenAI
 
-# Initialize the OpenAI client
-client = OpenAI(
-    base_url="https://api.studio.nebius.ai/v1/",
-    api_key=os.environ.get("NEBIUS_API_KEY")  # Make sure to set this in Hugging Face Secrets
-)
-
-# Streamlit app title
-st.title("AI Title Generator")
-
-# Text Area Input Bar
-user_input = st.text_area(
-    label="Enter a description for generating titles:",
-    placeholder="e.g., Man who went to jail for no reason"
-)
-
-# Generate Button
-if st.button("Generate Titles"):
-    if user_input.strip():
-        try:
-            # API call to OpenAI
-            completion = client.chat.completions.create(
-                model="nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
-                messages=[
-                    {"role": "system", "content": "Your task is to generate 3 very short titles based on the user input."},
-                    {"role": "user", "content": user_input}
-                ],
-                temperature=0.6,
-                maxTokens=512,
-                topP=0.9,
-                topK=50
-            )
-
-            # Extracting the generated titles
-            response_content = completion.choices[0].message["content"]
-
-            # Output Text Area
-            st.text_area(
-                label="Generated Titles:",
-                value=response_content,
-                height=200,
-                disabled=True
-            )
-
-        except Exception as e:
-            st.error(f"An error occurred: {e}")
-    else:
-        st.warning("Please provide input before clicking Generate.")
-
+# Set the API key
+API_KEY = os.getenv("NEBIUS_API_KEY")  # Use the secret key set in Hugging Face or locally
+if not API_KEY:
+    st.error("API Key is not set. Please set the NEBIUS_API_KEY environment variable.")
+else:
+    # Initialize the OpenAI client
+    client = OpenAI(
+        base_url="https://api.studio.nebius.ai/v1/",
+        api_key=API_KEY
+    )
+
+    # Streamlit app title
+    st.title("AI Title Generator")
+
+    # Text Area Input Bar
+    user_input = st.text_area(
+        label="Enter a description for generating titles:",
+        placeholder="e.g., Man who went to jail for no reason"
+    )
+
+    # Generate Button
+    if st.button("Generate Titles"):
+        if user_input.strip():
+            try:
+                # API call to OpenAI
+                completion = client.chat.completions.create(
+                    model="nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
+                    messages=[
+                        {"role": "system", "content": "Your task is to generate 3 very short titles based on the user input."},
+                        {"role": "user", "content": user_input}
+                    ],
+                    temperature=0.6,
+                    maxTokens=512,
+                    topP=0.9,
+                    topK=50
+                )
+
+                # Extracting the generated titles
+                response_content = completion.choices[0].message["content"]
+
+                # Output Text Area
+                st.text_area(
+                    label="Generated Titles:",
+                    value=response_content,
+                    height=200,
+                    disabled=True
+                )
+
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+        else:
+            st.warning("Please provide input before clicking Generate.")