Update app.py

app.py

@@ -1,151 +1,122 @@
-from pathlib import Path
-from typing import List, Dict, Tuple
-import matplotlib.colors as mpl_colors
-
-import pandas as pd
-import seaborn as sns
-import shinyswatch
-
-from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
-
-sns.set_theme()
-
-www_dir = Path(__file__).parent.resolve() / "www"
-
-df = pd.read_csv(Path(__file__).parent / "penguins.csv", na_values="NA")
-numeric_cols: List[str] = df.select_dtypes(include=["float64"]).columns.tolist()
-species: List[str] = df["Species"].unique().tolist()
-species.sort()
-
-app_ui = ui.page_fillable(
-    shinyswatch.theme.minty(),
-    ui.layout_sidebar(
-        ui.sidebar(
-            # Artwork by @allison_horst
-            ui.input_selectize(
-                "xvar",
-                "X variable",
-                numeric_cols,
-                selected="Bill Length (mm)",
-            ),
-            ui.input_selectize(
-                "yvar",
-                "Y variable",
-                numeric_cols,
-                selected="Bill Depth (mm)",
-            ),
-            ui.input_checkbox_group(
-                "species", "Filter by species", species, selected=species
-            ),
-            ui.hr(),
-            ui.input_switch("by_species", "Show species", value=True),
-            ui.input_switch("show_margins", "Show marginal plots", value=True),
-        ),
-        ui.output_ui("value_boxes"),
-        ui.output_plot("scatter", fill=True),
-        ui.help_text(
-            "Artwork by ",
-            ui.a("@allison_horst", href="https://twitter.com/allison_horst"),
-            class_="text-end",
-        ),
-    ),
-)
-
-
-def server(input: Inputs, output: Outputs, session: Session):
-    @reactive.Calc
-    def filtered_df() -> pd.DataFrame:
-        """Returns a Pandas data frame that includes only the desired rows"""
-
-        # This calculation "req"uires that at least one species is selected
-        req(len(input.species()) > 0)
-
-        # Filter the rows so we only include the desired species
-        return df[df["Species"].isin(input.species())]
-
-    @output
-    @render.plot
-    def scatter():
-        """Generates a plot for Shiny to display to the user"""
-
-        # The plotting function to use depends on whether margins are desired
-        plotfunc = sns.jointplot if input.show_margins() else sns.scatterplot
-
-        plotfunc(
-            data=filtered_df(),
-            x=input.xvar(),
-            y=input.yvar(),
-            palette=palette,
-            hue="Species" if input.by_species() else None,
-            hue_order=species,
-            legend=False,
-        )
-
-    @output
-    @render.ui
-    def value_boxes():
-        df = filtered_df()
-
-        def penguin_value_box(title: str, count: int, bgcol: str, showcase_img: str):
-            return ui.value_box(
-                title,
-                count,
-                {"class_": "pt-1 pb-0"},
-                showcase=ui.fill.as_fill_item(
-                    ui.tags.img(
-                        {"style": "object-fit:contain;"},
-                        src=showcase_img,
-                    )
-                ),
-                theme_color=None,
-                style=f"background-color: {bgcol};",
-            )
-
-        if not input.by_species():
-            return penguin_value_box(
-                "Penguins",
-                len(df.index),
-                bg_palette["default"],
-                # Artwork by @allison_horst
-                showcase_img="penguins.png",
-            )
-
-        value_boxes = [
-            penguin_value_box(
-                name,
-                len(df[df["Species"] == name]),
-                bg_palette[name],
-                # Artwork by @allison_horst
-                showcase_img=f"{name}.png",
-            )
-            for name in species
-            # Only include boxes for _selected_ species
-            if name in input.species()
-        ]
-
-        return ui.layout_column_wrap(*value_boxes, width = 1 / len(value_boxes))
-
-
-# "darkorange", "purple", "cyan4"
-colors = [[255, 140, 0], [160, 32, 240], [0, 139, 139]]
-colors = [(r / 255.0, g / 255.0, b / 255.0) for r, g, b in colors]
-
-palette: Dict[str, Tuple[float, float, float]] = {
-    "Adelie": colors[0],
-    "Chinstrap": colors[1],
-    "Gentoo": colors[2],
-    "default": sns.color_palette()[0],  # type: ignore
+import os
+from flask import Flask, request, jsonify
+from dotenv import load_dotenv
+from pymongo import MongoClient
+import openai
+from datetime import datetime
+
+load_dotenv()
+
+app = Flask(__name__)
+
+# MongoDB bağlantısı
+client = MongoClient(os.getenv("MONGODB_URI"))
+db = client[os.getenv("MONGODB_DB_NAME")]
+
+# OpenAI API anahtarı
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+# Model grupları ve modeller
+MODEL_GROUPS = {
+    "GPT4_GROUP": ["gpt-4", "gpt-4-0125-preview", "gpt-4-0613", "gpt-4-1106-preview"],
+    "GPT4_TURBO_GROUP": ["gpt-4-turbo", "gpt-4-turbo-preview"],
+    "GPT35_GROUP": ["gpt-3.5-turbo", "gpt-3.5-turbo-0125", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-1106"],
 }
 
-bg_palette = {}
-# Use `sns.set_style("whitegrid")` to help find approx alpha value
-for name, col in palette.items():
-    # Adjusted n_colors until `axe` accessibility did not complain about color contrast
-    bg_palette[name] = mpl_colors.to_hex(sns.light_palette(col, n_colors=7)[1])  # type: ignore
+MODELS = {
+    "GPT4": MODEL_GROUPS["GPT4_GROUP"] + MODEL_GROUPS["GPT4_TURBO_GROUP"],
+    "GPT35": MODEL_GROUPS["GPT35_GROUP"],
+    "IMAGE": ["dall-e-2", "dall-e-3"],
+}
 
+# Kullanıcı seviyeleri
+TIERS = {
+    1: {
+        "models": MODELS["GPT35"],
+        "limits": {"daily": 100, "GPT35_GROUP": 100}
+    },
+    2: {
+        "models": MODELS["GPT35"] + MODELS["GPT4"],
+        "limits": {"daily": 500, "GPT35_GROUP": 300, "GPT4_GROUP": 150, "GPT4_TURBO_GROUP": 50}
+    },
+    3: {
+        "models": MODELS["GPT35"] + MODELS["GPT4"] + MODELS["IMAGE"],
+        "limits": {"daily": 1000, "GPT35_GROUP": 500, "GPT4_GROUP": 300, "GPT4_TURBO_GROUP": 100, "dall-e-2": 50, "dall-e-3": 25}
+    }
+}
 
-app = App(
-    app_ui,
-    server,
-    static_assets=str(www_dir),
-)
+def get_user_tier(api_key):
+    user = db.users.find_one({"api_key": api_key})
+    if user:
+        return user["tier"]
+    for i in range(1, 4):
+        if os.getenv(f"USER_{i}_API_KEY") == api_key:
+            return int(os.getenv(f"USER_{i}_TIER"))
+    return None
+
+def get_model_group(model):
+    for group, models in MODEL_GROUPS.items():
+        if model in models:
+            return group
+    return model
+
+def check_rate_limit(api_key, tier, model):
+    now = datetime.now()
+    today = now.date().isoformat()
+    model_group = get_model_group(model)
+
+    usage = db.usage.find_one_and_update(
+        {"api_key": api_key, "date": today},
+        {"$inc": {"daily": 1, model_group: 1}},
+        upsert=True,
+        return_document=True
+    )
+
+    tier_limits = TIERS[tier]["limits"]
+    if usage["daily"] > tier_limits["daily"]:
+        raise Exception("Daily rate limit exceeded")
+    if usage.get(model_group, 0) > tier_limits.get(model_group, float("inf")):
+        raise Exception(f"Rate limit for {model_group} exceeded")
+
+@app.route('/v1/chat/completions', methods=['POST'])
+def chat_completions():
+    data = request.json
+    api_key = request.headers.get('Authorization', '').replace('Bearer ', '')
+    
+    user_tier = get_user_tier(api_key)
+    if not user_tier:
+        return jsonify({"error": "Invalid API key"}), 401
+
+    model = data.get('model')
+    if model not in TIERS[user_tier]["models"]:
+        return jsonify({"error": "Model not available for your tier"}), 403
+
+    try:
+        check_rate_limit(api_key, user_tier, model)
+        response = openai.ChatCompletion.create(**data)
+        return jsonify(response)
+    except Exception as e:
+        return jsonify({"error": str(e)}), 429
+
+@app.route('/v1/images/generations', methods=['POST'])
+def image_generations():
+    data = request.json
+    api_key = request.headers.get('Authorization', '').replace('Bearer ', '')
+    
+    user_tier = get_user_tier(api_key)
+    if not user_tier:
+        return jsonify({"error": "Invalid API key"}), 401
+
+    model = data.get('model', 'dall-e-2')
+    if model not in TIERS[user_tier]["models"]:
+        return jsonify({"error": "Model not available for your tier"}), 403
+
+    try:
+        check_rate_limit(api_key, user_tier, model)
+        response = openai.Image.create(**data)
+        return jsonify(response)
+    except Exception as e:
+        return jsonify({"error": str(e)}), 429
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)