Update main.py

main.py

@@ -19,7 +19,7 @@ app.add_middleware(  # add the middleware
 
 google_api_key = os.environ["GOOGLE_API_KEY"]
 cse_id = os.environ["GOOGLE_CSE_ID"]
-model = os.environ['default_model']
+
 
 def search_google(query):
     service = build("customsearch", "v1", developerKey=google_api_key)
@@ -35,15 +35,20 @@ def hello():
     return "Hello! My name is Linlada."
 
 
+def llm(prompt, model):
+    llm = G4F(model=model)
+    response = llm(prompt)
+    return response
+
 
-def gpt_with_google_search(prompt):
+def gpt_with_google_search(prompt, model):
     search_results = search_google(prompt)
     text = ""
     ref = ""
     for item in search_results:
         text += item['title'] + "\n" + item['snippet'] + "\n\n"
         ref += "- {} ({})\n".format(item['title'], item['link'])
-    results = generate_text(f'Summarize: {text}')
+    results = llm(prompt = f'Summarize: {text}', model = model)
     res = "{} \n\n {}".format(results, ref)
     return res
 
@@ -63,8 +68,7 @@ def gpt_with_google_search(prompt):
 class Linlada(BaseModel):
     prompt: str
     web_access: str = True
-    model: str = 'gpt-3.5-turbo'
-
+    model: str = os.environ['default_model']
 
 
 @app.post('/linlada')
@@ -72,11 +76,10 @@ def linlada(request: Linlada):
     prompt = request.prompt
     model = request.model
     web_access = request.web_access
-    llm = G4F(model=model)
     if web_access == "True":
-        chat = gpt_with_google_search(prompt)
+        chat = gpt_with_google_search(prompt, model)
     else:
-        chat = llm(prompt)
+        chat = llm(prompt, model)
     return chat