Upload 3 files

app.py

@@ -0,0 +1,89 @@
+# coding: utf-8
+
+import gradio as gr
+from theme import CustomTheme
+import pandas as pd
+import time
+import openai
+import os
+
+
+custom_theme = CustomTheme()
+
+
+openai.api_key = os.environ.get("OPENAI_API_KEY")
+
+ft_model_ada = 'ada:ft-personal-2023-05-17-17-03-33'
+ft_model_davinci =  'ada:ft-personal-2023-05-17-17-03-33'
+
+def upload_file(file, model_name):
+
+    with open(file.name, "r", encoding='unicode_escape') as f:
+        content = f.read().replace('\n', '')
+
+    sentences = list(map(str.strip, content.split(".")))
+    df = pd.DataFrame({"claim": sentences}, index=range(len(sentences)))
+    # df = pd.DataFrame({"claim": sentences[:10]}, index=range(len(sentences[:10])))
+
+    col = f"openai_pred_{model_name}"
+    model = ft_model_ada if model_name == "ada" else ft_model_davinci
+
+    results = []
+
+    start_time = time.time()
+
+    for i in range(len(sentences)):
+        res = openai.Completion.create(model=model, prompt= sentences[i] + ' ->', max_tokens=1, temperature=0)
+        results.append(res['choices'][0]['text'])
+
+    end_time = time.time()
+
+    # results = sentences[:10]
+
+    df[col] = results
+    df[col] = df[col].map({" True": 1, " False": 0})
+
+    time_taken = round(end_time - start_time, 2)
+
+    display_df_openai_ada = df.loc[df[col] == 1]
+
+    return display_df_openai_ada, time_taken
+
+    
+def export_csv(d, file):
+    filename = file.name.split("/")[-1].split(".")[0]
+    d.to_csv(f"{filename}_output.csv")
+    # filename = file.name.split(".")[0]
+    return gr.File.update(value = f"{filename}_output.csv", visible=True)
+
+with gr.Blocks(theme=custom_theme) as demo:
+
+    gr.Markdown(value="# Upload transcription txt file")
+
+    with gr.Row():
+        input_document_pdf = gr.File(label="Uplaod transcription file")
+
+    with gr.Row():
+
+        with gr.Column():
+            model_name_ada = gr.Text(label="model_name", value="ada", visible=False)
+            output_df = gr.Dataframe(row_count = (2, "fixed"), col_count=(2, "fixed"), headers=["claim", "prediction"],
+                                    label="Predictions", max_rows=3, max_cols=2, wrap= True)
+
+            time_taken = gr.Text(label="Time Taken", lines=1)
+            download = gr.Button("Download output", variant="primary")
+
+            csv = gr.File(interactive=False, visible=False)
+
+    input_document_pdf.upload(
+        upload_file,
+        [input_document_pdf, model_name_ada],
+        [output_df, time_taken]
+    )
+
+    download.click(export_csv, [output_df, input_document_pdf], [csv])
+
+
+
+if __name__ == "__main__":
+    demo.launch(debug=True)

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+openai
+pandas

theme.py

@@ -0,0 +1,95 @@
+from __future__ import annotations
+from typing import Iterable
+import gradio as gr
+from gradio.themes.base import Base
+from gradio.themes import Soft
+from gradio.themes.utils import colors, fonts, sizes
+
+
+class CustomTheme(Base):
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.blue,
+        secondary_hue: colors.Color | str = colors.cyan,
+        neutral_hue: colors.Color | str = colors.zinc,
+        spacing_size: sizes.Size | str = sizes.spacing_md,
+        radius_size: sizes.Size | str = sizes.radius_md,
+        text_size: sizes.Size | str = sizes.text_md,
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_hue,
+            neutral_hue=neutral_hue,
+            spacing_size=spacing_size,
+            radius_size=radius_size,
+            text_size=text_size,
+        )
+        self.name = "custom_theme"
+        super().set(
+            # Colors
+            background_fill_primary="*neutral_50",
+            slider_color="*primary_500",
+            slider_color_dark="*primary_600",
+            # Shadows
+            shadow_drop="0 1px 4px 0 rgb(0 0 0 / 0.1)",
+            shadow_drop_lg="0 2px 5px 0 rgb(0 0 0 / 0.1)",
+            # Block Labels
+            block_background_fill="white",
+            block_label_padding="*spacing_sm *spacing_md",
+            block_label_background_fill="*primary_100",
+            block_label_background_fill_dark="*primary_600",
+            block_label_radius="*radius_md",
+            block_label_text_size="*text_md",
+            block_label_text_weight="600",
+            block_label_text_color="*primary_500",
+            block_label_text_color_dark="*white",
+            block_title_radius="*block_label_radius",
+            block_title_padding="*block_label_padding",
+            block_title_background_fill="*block_label_background_fill",
+            block_title_text_weight="600",
+            block_title_text_color="*primary_500",
+            block_title_text_color_dark="*white",
+            block_label_margin="*spacing_md",
+            block_shadow="*shadow_drop_lg",
+            # Inputs
+            input_border_color="*neutral_50",
+            input_shadow="*shadow_drop",
+            input_shadow_focus="*shadow_drop_lg",
+            checkbox_shadow="none",
+            # Buttons
+            shadow_spread="6px",
+            button_shadow="*shadow_drop_lg",
+            button_shadow_hover="*shadow_drop_lg",
+            button_shadow_active="*shadow_inset",
+            button_primary_background_fill="linear-gradient(90deg, *primary_300, *secondary_400)",
+            button_primary_background_fill_hover="linear-gradient(90deg, *primary_200, *secondary_300)",
+            button_primary_text_color="white",
+            button_primary_background_fill_dark="linear-gradient(90deg, *primary_600, *secondary_800)",
+            button_primary_background_fill_hover_dark="*primary_500",
+            button_secondary_background_fill="white",
+            button_secondary_background_fill_hover="*neutral_100",
+            button_secondary_background_fill_hover_dark="*primary_500",
+            button_secondary_text_color="*neutral_800",
+            button_cancel_background_fill="*button_secondary_background_fill",
+            button_cancel_background_fill_hover="*button_secondary_background_fill_hover",
+            button_cancel_background_fill_hover_dark="*button_secondary_background_fill_hover",
+            button_cancel_text_color="*button_secondary_text_color",
+            # checkboxes
+            checkbox_label_shadow="*shadow_drop_lg",
+            checkbox_label_background_fill_selected="*primary_500",
+            checkbox_label_background_fill_selected_dark="*primary_600",
+            checkbox_border_width="1px",
+            checkbox_border_color="*neutral_100",
+            checkbox_border_color_dark="*neutral_600",
+            checkbox_background_color_selected="*primary_600",
+            checkbox_background_color_selected_dark="*primary_700",
+            checkbox_border_color_focus="*primary_500",
+            checkbox_border_color_focus_dark="*primary_600",
+            checkbox_border_color_selected="*primary_600",
+            checkbox_border_color_selected_dark="*primary_700",
+            checkbox_label_text_color_selected="white",
+            # Borders
+            block_border_width="0px",
+            panel_border_width="1px",
+        )