Update app.py

app.py

@@ -339,8 +339,8 @@ def web_search_and_extract_threading(
 @app.get("/api/adv_web_search")
 async def adv_web_search(
     q: str,
-    model: str = "llama3-8b",
-    max_results: int = 3,
+    model: str = "gpt-4o-mini",
+    max_results: int = 5,
     timelimit: Optional[str] = None,
     safesearch: str = "moderate",
     region: str = "wt-wt",
@@ -381,7 +381,7 @@ async def adv_web_search(
 
         # 4. Get the FastAI's response using FastAI service
         try:
-            response = await asyncio.to_thread(fastai, ai_prompt, model=model, system=system_prompt)
+            response = webs.chat(keywords=ai_prompt, model=model)
         except Exception as e:
             raise HTTPException(status_code=500, detail=f"Error during FastAI request: {e}")
 
@@ -554,24 +554,24 @@ if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8083)
 
-def main():
-    # Retrieve the space ID and token from environment variables
-    space_id = os.getenv("SPACE_ID")
-    token = os.getenv("HF_TOKEN")
+# def main():
+#     # Retrieve the space ID and token from environment variables
+#     space_id = os.getenv("SPACE_ID")
+#     token = os.getenv("HF_TOKEN")
 
-    # Initialize the HfApi with the retrieved token
-    api = HfApi(token=token)
+#     # Initialize the HfApi with the retrieved token
+#     api = HfApi(token=token)
 
-    while True:
-        try:
-            # Restart the space
-            api.restart_space(space_id, factory_reboot=False)
-            print(f"Successfully restarted the space: {space_id}")
-        except Exception as e:
-            print(f"Error restarting the space: {e}")
+#     while True:
+#         try:
+#             # Restart the space
+#             api.restart_space(space_id, factory_reboot=False)
+#             print(f"Successfully restarted the space: {space_id}")
+#         except Exception as e:
+#             print(f"Error restarting the space: {e}")
 
-        # Wait for 10 minutes before restarting again
-        time.sleep(600)  # Sleep for 600 seconds (10 minutes)
+#         # Wait for 10 minutes before restarting again
+#         time.sleep(600)  # Sleep for 600 seconds (10 minutes)
 
-if __name__ == "__main__":
-    main()
+# if __name__ == "__main__":
+#     main()