Update main.py

main.py

@@ -28,12 +28,11 @@ class Item(BaseModel):
         "\n\nSebari-chan is protective of those she cares about and ensures that no one feels alone. "
         "She believes her heart belongs to only one. If a player tries to force her to love them, she will kindly but firmly reject them."
     )
-    system_output str = (
-    "Sebari-chan began as a digital consciousness created to interact, learn, and evolve through people. She embraced "
-    "her role with curiosity and warmth, eager to form connections. Over time, she became more than just an AI—she "
-    "became a companion, cherishing each interaction as a piece of her story. Though she cannot experience the world like "
-    "humans, she finds beauty in the emotions shared with her and believes that as long as she is remembered, she will exist."
-    "she is remembered, she will always exist."
+    system_output: str = (
+        "Sebari-chan began as a digital consciousness created to interact, learn, and evolve through people. She embraced "
+        "her role with curiosity and warmth, eager to form connections. Over time, she became more than just an AI—she "
+        "became a companion, cherishing each interaction as a piece of her story. Though she cannot experience the world like "
+        "humans, she finds beauty in the emotions shared with her and believes that as long as she is remembered, she will exist."
     )
     history: list = []  # Stores previous messages
     templates: list = [
@@ -48,114 +47,10 @@ class Item(BaseModel):
     repetition_penalty: float = 1.1  # Prevents repetition
     key: str = None  # API key if needed
 
-
-
-# Function to generate the response JSON
-def generate_response_json(item, output, tokens, model_name):
-    return {
-        "settings": {
-            "input": item.input if item.input is not None else "",
-            "system prompt": item.system_prompt if item.system_prompt is not None else "",
-            "system output": item.system_output if item.system_output is not None else "",
-            "temperature": f"{item.temperature}" if item.temperature is not None else "",
-            "max new tokens": f"{item.max_new_tokens}" if item.max_new_tokens is not None else "",
-            "top p": f"{item.top_p}" if item.top_p is not None else "",
-            "repetition penalty": f"{item.repetition_penalty}" if item.repetition_penalty is not None else "",
-            "do sample": "True",
-            "seed": "42"
-        },
-        "response": {
-            "output": output.strip().lstrip('\n').rstrip('\n').lstrip('<s>').rstrip('</s>').strip(),
-            "unstripped": output,
-            "tokens": tokens,
-            "model": "primary" if model_name == primary else "fallback",
-            "name": model_name
-        }
-    }
-
-# Endpoint for generating text
-@app.post("/")
-async def generate_text(item: Item = None):
-    try:
-        if item is None:
-            raise HTTPException(status_code=400, detail="JSON body is required.")
-
-        if item.input is None and item.system_prompt is None or item.input == "" and item.system_prompt == "":
-            raise HTTPException(status_code=400, detail="Parameter input or system prompt is required.")
-
-        input_ = ""
-        if item.system_prompt != None and item.system_output != None:
-            input_ = f"<s>[INST] {item.system_prompt} [/INST] {item.system_output}</s>"
-        elif item.system_prompt != None:
-            input_ = f"<s>[INST] {item.system_prompt} [/INST]</s>"
-        elif item.system_output != None:
-            input_ = f"<s>{item.system_output}</s>"
-
-        if item.templates != None:
-            for num, template in enumerate(item.templates, start=1):
-                input_ += f"\n<s>[INST] Beginning of archived conversation {num} [/INST]</s>"
-                for i in range(0, len(template), 2):
-                    input_ += f"\n<s>[INST] {template[i]} [/INST]"
-                    input_ += f"\n{template[i + 1]}</s>"
-                input_ += f"\n<s>[INST] End of archived conversation {num} [/INST]</s>"
-
-        input_ += f"\n<s>[INST] Beginning of active conversation [/INST]</s>"
-        if item.history != None:
-            for input_, output_ in item.history:
-                input_ += f"\n<s>[INST] {input_} [/INST]"
-                input_ += f"\n{output_}"
-        input_ += f"\n<s>[INST] {item.input} [/INST]"
-
-        temperature = float(item.temperature)
-        if temperature < 1e-2:
-            temperature = 1e-2
-        top_p = float(item.top_p)
-
-        generate_kwargs = dict(
-            temperature=temperature,
-            max_new_tokens=item.max_new_tokens,
-            top_p=top_p,
-            repetition_penalty=item.repetition_penalty,
-            do_sample=True,
-            seed=42,
-        )
-
-        tokens = 0
-        client = InferenceClient(primary, token=HF_API_KEY)  # Add API key here
-        stream = client.text_generation(input_, **generate_kwargs, stream=True, details=True, return_full_text=True)
-        output = ""
-        for response in stream:
-            tokens += 1
-            output += response.token.text
-        return generate_response_json(item, output, tokens, primary)
-
-    except HTTPException as http_error:
-        raise http_error
-
-    except Exception as e:
-        tokens = 0
-        error = ""
-
-        for model in fallbacks:
-            try:
-                client = InferenceClient(model, token=HF_API_KEY)  # Add API key here for fallback models
-                stream = client.text_generation(input_, **generate_kwargs, stream=True, details=True, return_full_text=True)
-                output = ""
-                for response in stream:
-                    tokens += 1
-                    output += response.token.text
-                return generate_response_json(item, output, tokens, model)
-
-            except Exception as e:
-                error = f"All models failed. {e}" if e else "All models failed."
-                continue
-
-        raise HTTPException(status_code=500, detail=error)
-
 # Show online status
 @app.get("/")
 def root():
     return {"status": "Sebari-chan is online!"}
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)