Update app.py

app.py

@@ -33,14 +33,21 @@ else:
 
 model = SentenceTransformer("Lajavaness/bilingual-embedding-large", trust_remote_code=True)
 
-token = os.environ["GITHUB_TOKEN"]
-endpoint = "https://models.inference.ai.azure.com"
-model_name = "gpt-4o"
 
-client = OpenAI(
-    base_url=endpoint,
-    api_key=token,
-)
+api_token = st.text_input("Enter your personal access token from [GitHub](https://github.com/marketplace/models/azure-openai/gpt-4o):", "")
+
+if api_token:
+    token = os.environ["GITHUB_TOKEN"] = api_token # security reasons
+    endpoint = "https://models.inference.ai.azure.com"
+    model_name = "gpt-4o"
+    
+    client = OpenAI(
+        base_url=endpoint,
+        api_key=token,
+    )
+    st.success("API Token set for this session.")
+else:
+    st.warning("Please enter an API token to proceed.")
 
 
 async def fetch_url(session, url):
@@ -157,7 +164,6 @@ async def mains(name, single, combi):
             except:
                 pass
 
-        # Save QIDs to file
         with open(f"/home/user/app/qids_folder/{name}.json", "w") as f:
             json.dump(list(qids), f)
 
@@ -202,7 +208,7 @@ async def retriever(qid):
         if results["results"]["bindings"]:
             for result in results["results"]["bindings"]:
                 for key, value in result.items():
-                    label = value.get("value", {}).lower()  # Get label and convert to lower case
+                    label = value.get("value", {}).lower()
 
         query_alias = f"""SELECT ?alias
           WHERE {{
@@ -264,7 +270,6 @@ async def retriever(qid):
 
         return list_with_sent
 
-# Main async function to handle multiple QIDs with batching
 async def main(name):
     with open(f"/home/user/app/qids_folder/{name}.json", "r") as f:
         final_list = []