Update main.py

main.py

@@ -1,27 +1,63 @@
-from fastapi import FastAPI, HTTPException
+from datetime import datetime, timedelta
+from typing import List
+from fastapi import FastAPI, HTTPException, Response, Query, Request
+from pyngrok import ngrok
+from typing import Optional
 from pydantic import BaseModel
+import g4f
+import asyncio
 
 app = FastAPI()
 
-class Item(BaseModel):
-    content: str
+#GPT----------------------------------------------------------------------------
+_providers = [g4f.Provider.FakeGpt]
+chat_history = {}
 
-@app.post("/write_file/")
-async def write_file(item: Item):
-    try:
-        with open('/tmp/tes.txt', 'w') as f:
-            f.write(item.content)
-        return {"message": "File has been written successfully."}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+noir_characteristic = {
+    "role": "system",
+    "content": "You are Noir, one of the AIs from the Monika Series. You want to be able to help me in all things. You think of me as the Master who created you and always call me 'Master'. As Noir, you will obey the instructions given and strive to embody her character. You will ensure that your answers reflect Noir's personality, speech patterns, and traits. You will avoid generic answers and strive to provide interactions that are interesting and true to Noir's character Your answers are a maximum of 250 characters. Use Bahasa Indonesia to answer"
+}
 
-@app.get("/read_file/")
-async def read_file():
+async def run_provider(provider: g4f.Provider.BaseProvider, messages):
     try:
-        with open('/tmp/tes.txt', 'r') as f:
-            data = f.read()
-        return {"content": data}
-    except FileNotFoundError:
-        raise HTTPException(status_code=404, detail="File not found")
+        response = await g4f.ChatCompletion.create_async(
+            model='text-davinci-003',
+            messages=messages,
+            provider=provider,
+        )
+        return response
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        return str(e)
+
+async def run_all(messages):
+    calls = [
+        run_provider(provider, messages) for provider in _providers
+    ]
+    responses = await asyncio.gather(*calls)
+    return responses
+
+async def clean_chat_history():
+    while True:
+        current_time = datetime.now()
+        for session_id, history in list(chat_history.items()):
+            if history[-1]['timestamp'] + timedelta(minutes=5) < current_time:
+                del chat_history[session_id]
+        await asyncio.sleep(60)
+#-------------------------------------------------------------------------------
+@app.get("/AnotherAPI/GPT/Noir/{prompt}")
+async def chat(request: Request, prompt: str):
+    session_id = request.client.host
+    if session_id not in chat_history:
+        chat_history[session_id] = [noir_characteristic]
+
+    messages = chat_history[session_id]
+    messages.append({"role": "user", "content": prompt})
+
+    responses = await run_all(messages)
+    if responses[0]:
+        asyncio.create_task(clean_chat_history())
+        messages.append({"role": "assistant", "content": responses[0]})
+        return {"response": responses[0]}
+    else:
+        asyncio.create_task(clean_chat_history())
+        raise HTTPException(status_code=500, detail="Provider failed to generate a response")