Initial commit

main.py

@@ -1,11 +1,12 @@
 #!/bin/env python3.11
+import gradio as gr
 import os
 import sqlite3
 import replicate
 import argparse
 import requests
 from datetime import datetime
-from fastapi import FastAPI, WebSocket, WebSocketDisconnect, HTTPException, Request, Form, Query, Response
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect, HTTPException, Request, Form, Query,Response
 from fastapi.templating import Jinja2Templates
 from fastapi.responses import FileResponse
 from fastapi.staticfiles import StaticFiles
@@ -14,20 +15,17 @@ from typing import Optional, List
 import uvicorn
 from asyncio import gather, Semaphore, create_task
 from mistralai import Mistral
-from dotenv import load_dotenv
 from contextlib import contextmanager
 from io import BytesIO
 import zipfile
-import logging
-
-# Logging-Konfiguration (gut!)
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("uvicorn.error")
 
 import sys
 print(f"Arguments: {sys.argv}")
 
-load_dotenv()
+
+token = os.getenv("HF_TOKEN")
+api_key = os.getenv("MISTRAL_API_KEY")
+agent_id = os.getenv("MISTRAL_FLUX_AGENT")
 
 # ANSI Escape Codes für farbige Ausgabe (kann entfernt werden, falls nicht benötigt)
 HEADER = "\033[38;2;255;255;153m"
@@ -707,9 +705,9 @@ if __name__ == "__main__":
 
     # Run the Uvicorn server
     uvicorn.run(
-        "main:app",
+        "app:app",
         host="0.0.0.0",
-        port=8000,
+        port=7860,
         reload=True,
         log_level="trace"
     )