Upload 2 files

app.py

@@ -0,0 +1,108 @@
+import os
+import gradio as gr
+import asyncio
+import pandas as pd
+from datetime import datetime
+from langchain.schema.output_parser import StrOutputParser
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts.chat import (
+    ChatPromptTemplate,
+    HumanMessagePromptTemplate,
+    SystemMessagePromptTemplate,
+)
+
+# Load OpenAI model with API key
+openai_key = os.getenv('OPENAI_API_KEY')
+model = ChatOpenAI(model_name="gpt-4-0125-preview", temperature=0.4)
+
+# System prompt templates
+raft_base_template  = os.getenv('RAFT_BASE_TEMPLATE')
+
+# System prompt template
+customer_template = os.getenv('CUSTOMER_TEMPLATE')
+
+raft_metaprompt_template = os.getenv('RAFT_METAPROMPT_TEMPLATE')
+# Initialize DataFrame
+results_df = pd.DataFrame(columns=['Experiment Number','Temperature', 'PromptID','Business Idea', 'Generated Domains'])
+
+def extract_content_inside_brackets(json_string):
+    # Находим позицию первой открывающей и последней закрывающей скобки
+    start = json_string.find('[')
+    end = json_string.find(']', start)
+
+    # Если скобки найдены, извлекаем содержимое между ними
+    if start != -1 and end != -1:
+        content_inside_brackets = json_string[start+1:end]
+        return content_inside_brackets
+    else:
+        # Возвращаем None, если скобки не найдены
+        return None
+list_of_prompts = ['customer_template', 'raft_base_template', 'raft_metaprompt_template']
+# Function to generate domain names based on different templates
+async def generate_domain_names(business_idea, temperature):
+    model.temperature = temperature  # Update model temperature
+    prompts = [customer_template, raft_base_template, raft_metaprompt_template]
+    tasks = []
+    for prompt in prompts:
+        system_message_prompt = SystemMessagePromptTemplate.from_template(prompt)
+        human_message_prompt = HumanMessagePromptTemplate.from_template("User's input: ~###~ {user_input}~###~,")
+        combined_prompt = ChatPromptTemplate.from_messages([system_message_prompt, human_message_prompt])
+        json_chain = combined_prompt | model | StrOutputParser() | extract_content_inside_brackets
+        task = asyncio.to_thread(json_chain.invoke, {"user_input": business_idea})
+        tasks.append(task)
+    domains = await asyncio.gather(*tasks)
+    # Update DataFrame and save to JSON
+    experiment_number = datetime.now().strftime("%Y%m%d%H%M%S")
+    for i, domain in enumerate(domains):
+        new_row = {
+            'Experiment Number': experiment_number,
+            'Business Idea': business_idea,
+            'Temperature': temperature,
+            'PromptID': list_of_prompts[i],
+            'Generated Domains': domain
+        }
+        global results_df
+        results_df = pd.concat([results_df, pd.DataFrame([new_row])], ignore_index=True)
+    save_to_json(results_df)
+    return domains
+
+# Function to save DataFrame to JSON
+def save_to_json(df):
+    json_path = 'domain_name_generation_results.json'
+    df.to_json(json_path, orient='records', lines=True)
+
+# Function to download the DataFrame as an Excel file
+def download_excel():
+    df = pd.read_json('domain_name_generation_results.json', lines=True)
+    excel_path = 'domain_name_generation_results.xlsx'
+    df.to_excel(excel_path, index=False)
+    return excel_path
+
+# Setup Gradio interface
+def setup_interface():
+    with gr.Blocks() as iface:
+        with gr.Row():
+            text_input = gr.Textbox(label="Enter your business idea", placeholder="Type here...", lines=2)
+            temperature_slider = gr.Slider(minimum=0, maximum=1, step=0.01, label="Set Temperature")
+            submit_button = gr.Button("Generate")
+        with gr.Row():
+            output1 = gr.Textbox(label="Output for Customer Template")
+            output2 = gr.Textbox(label="Output for Base Raft Template")
+            output3 = gr.Textbox(label="Output for MetaPrompt Raft Template")
+        with gr.Row():
+            download_btn = gr.Button("Download Excel")
+        submit_button.click(
+            fn=generate_domain_names,
+            inputs=[text_input, temperature_slider],
+            outputs=[output1, output2, output3]
+        )
+        download_btn.click(
+            fn=download_excel,
+            inputs=None,
+            outputs=gr.File(label="Download Excel")
+        )
+    return iface
+
+
+iface = setup_interface()
+iface.launch()

requirements.txt

@@ -0,0 +1,173 @@
+aiofiles==23.2.1 ; python_version >= "3.10" and python_version < "4.0"
+aiohttp-retry==2.8.3 ; python_version >= "3.10" and python_version < "4.0"
+aiohttp==3.9.5 ; python_version >= "3.10" and python_version < "4.0"
+aiosignal==1.3.1 ; python_version >= "3.10" and python_version < "4.0"
+altair==5.3.0 ; python_version >= "3.10" and python_version < "4.0"
+amqp==5.2.0 ; python_version >= "3.10" and python_version < "4.0"
+annotated-types==0.6.0 ; python_version >= "3.10" and python_version < "4.0"
+antlr4-python3-runtime==4.9.3 ; python_version >= "3.10" and python_version < "4.0"
+anyio==4.3.0 ; python_version >= "3.10" and python_version < "4.0"
+appdirs==1.4.4 ; python_version >= "3.10" and python_version < "4.0"
+appnope==0.1.4 ; python_version >= "3.10" and python_version < "4.0" and platform_system == "Darwin"
+asttokens==2.4.1 ; python_version >= "3.10" and python_version < "4.0"
+asyncssh==2.14.2 ; python_version >= "3.10" and python_version < "4.0"
+atpublic==4.1.0 ; python_version >= "3.10" and python_version < "4.0"
+attrs==23.2.0 ; python_version >= "3.10" and python_version < "4.0"
+billiard==4.2.0 ; python_version >= "3.10" and python_version < "4.0"
+celery==5.4.0 ; python_version >= "3.10" and python_version < "4.0"
+certifi==2024.2.2 ; python_version >= "3.10" and python_version < "4.0"
+cffi==1.16.0 ; python_version >= "3.10" and python_version < "4.0"
+charset-normalizer==3.3.2 ; python_version >= "3.10" and python_version < "4.0"
+click-didyoumean==0.3.1 ; python_version >= "3.10" and python_version < "4.0"
+click-plugins==1.1.1 ; python_version >= "3.10" and python_version < "4.0"
+click-repl==0.3.0 ; python_version >= "3.10" and python_version < "4.0"
+click==8.1.7 ; python_version >= "3.10" and python_version < "4.0"
+colorama==0.4.6 ; python_version >= "3.10" and python_version < "4.0"
+comm==0.2.2 ; python_version >= "3.10" and python_version < "4.0"
+configobj==5.0.8 ; python_version >= "3.10" and python_version < "4.0"
+contourpy==1.2.1 ; python_version >= "3.10" and python_version < "4.0"
+cryptography==42.0.5 ; python_version >= "3.10" and python_version < "4.0"
+cycler==0.12.1 ; python_version >= "3.10" and python_version < "4.0"
+dataclasses-json==0.6.4 ; python_version >= "3.10" and python_version < "4.0"
+debugpy==1.8.1 ; python_version >= "3.10" and python_version < "4.0"
+decorator==5.1.1 ; python_version >= "3.10" and python_version < "4.0"
+dictdiffer==0.9.0 ; python_version >= "3.10" and python_version < "4.0"
+diskcache==5.6.3 ; python_version >= "3.10" and python_version < "4.0"
+distro==1.9.0 ; python_version >= "3.10" and python_version < "4.0"
+dpath==2.1.6 ; python_version >= "3.10" and python_version < "4.0"
+dulwich==0.21.7 ; python_version >= "3.10" and python_version < "4.0"
+dvc-data==3.15.1 ; python_version >= "3.10" and python_version < "4.0"
+dvc-http==2.32.0 ; python_version >= "3.10" and python_version < "4.0"
+dvc-objects==5.1.0 ; python_version >= "3.10" and python_version < "4.0"
+dvc-render==1.0.2 ; python_version >= "3.10" and python_version < "4.0"
+dvc-studio-client==0.20.0 ; python_version >= "3.10" and python_version < "4.0"
+dvc-task==0.4.0 ; python_version >= "3.10" and python_version < "4.0"
+dvc==3.50.0 ; python_version >= "3.10" and python_version < "4.0"
+entrypoints==0.4 ; python_version >= "3.10" and python_version < "4.0"
+et-xmlfile==1.1.0 ; python_version >= "3.10" and python_version < "4.0"
+executing==2.0.1 ; python_version >= "3.10" and python_version < "4.0"
+fastapi==0.110.2 ; python_version >= "3.10" and python_version < "4.0"
+ffmpy==0.3.2 ; python_version >= "3.10" and python_version < "4.0"
+filelock==3.13.4 ; python_version >= "3.10" and python_version < "4.0"
+flatten-dict==0.4.2 ; python_version >= "3.10" and python_version < "4.0"
+flufl-lock==7.1.1 ; python_version >= "3.10" and python_version < "4.0"
+fonttools==4.51.0 ; python_version >= "3.10" and python_version < "4.0"
+frozenlist==1.4.1 ; python_version >= "3.10" and python_version < "4.0"
+fsspec==2024.3.1 ; python_version >= "3.10" and python_version < "4.0"
+fsspec[http]==2024.3.1 ; python_version >= "3.10" and python_version < "4.0"
+fsspec[tqdm]==2024.3.1 ; python_version >= "3.10" and python_version < "4.0"
+funcy==2.0 ; python_version >= "3.10" and python_version < "4.0"
+gitdb==4.0.11 ; python_version >= "3.10" and python_version < "4.0"
+gitpython==3.1.43 ; python_version >= "3.10" and python_version < "4.0"
+gradio-client==0.15.1 ; python_version >= "3.10" and python_version < "4.0"
+gradio==4.27.0 ; python_version >= "3.10" and python_version < "4.0"
+grandalf==0.8 ; python_version >= "3.10" and python_version < "4.0"
+greenlet==3.0.3 ; python_version >= "3.10" and python_version < "4.0" and (platform_machine == "win32" or platform_machine == "WIN32" or platform_machine == "AMD64" or platform_machine == "amd64" or platform_machine == "x86_64" or platform_machine == "ppc64le" or platform_machine == "aarch64")
+gto==1.7.1 ; python_version >= "3.10" and python_version < "4.0"
+h11==0.14.0 ; python_version >= "3.10" and python_version < "4.0"
+httpcore==1.0.5 ; python_version >= "3.10" and python_version < "4.0"
+httpx==0.27.0 ; python_version >= "3.10" and python_version < "4.0"
+huggingface-hub==0.22.2 ; python_version >= "3.10" and python_version < "4.0"
+hydra-core==1.3.2 ; python_version >= "3.10" and python_version < "4.0"
+idna==3.7 ; python_version >= "3.10" and python_version < "4.0"
+importlib-resources==6.4.0 ; python_version >= "3.10" and python_version < "4.0"
+ipykernel==6.29.4 ; python_version >= "3.10" and python_version < "4.0"
+ipython==8.23.0 ; python_version >= "3.10" and python_version < "4.0"
+iterative-telemetry==0.0.8 ; python_version >= "3.10" and python_version < "4.0"
+jedi==0.19.1 ; python_version >= "3.10" and python_version < "4.0"
+jinja2==3.1.3 ; python_version >= "3.10" and python_version < "4.0"
+jsonpatch==1.33 ; python_version >= "3.10" and python_version < "4.0"
+jsonpointer==2.4 ; python_version >= "3.10" and python_version < "4.0"
+jsonschema-specifications==2023.12.1 ; python_version >= "3.10" and python_version < "4.0"
+jsonschema==4.21.1 ; python_version >= "3.10" and python_version < "4.0"
+jupyter-client==8.6.1 ; python_version >= "3.10" and python_version < "4.0"
+jupyter-core==5.7.2 ; python_version >= "3.10" and python_version < "4.0"
+kiwisolver==1.4.5 ; python_version >= "3.10" and python_version < "4.0"
+kombu==5.3.7 ; python_version >= "3.10" and python_version < "4.0"
+langchain-community==0.0.33 ; python_version >= "3.10" and python_version < "4.0"
+langchain-core==0.1.44 ; python_version >= "3.10" and python_version < "4.0"
+langchain-text-splitters==0.0.1 ; python_version >= "3.10" and python_version < "4.0"
+langchain==0.1.16 ; python_version >= "3.10" and python_version < "4.0"
+langsmith==0.1.48 ; python_version >= "3.10" and python_version < "4.0"
+markdown-it-py==3.0.0 ; python_version >= "3.10" and python_version < "4.0"
+markupsafe==2.1.5 ; python_version >= "3.10" and python_version < "4.0"
+marshmallow==3.21.1 ; python_version >= "3.10" and python_version < "4.0"
+matplotlib-inline==0.1.7 ; python_version >= "3.10" and python_version < "4.0"
+matplotlib==3.8.4 ; python_version >= "3.10" and python_version < "4.0"
+mdurl==0.1.2 ; python_version >= "3.10" and python_version < "4.0"
+multidict==6.0.5 ; python_version >= "3.10" and python_version < "4.0"
+mypy-extensions==1.0.0 ; python_version >= "3.10" and python_version < "4.0"
+nest-asyncio==1.6.0 ; python_version >= "3.10" and python_version < "4.0"
+networkx==3.3 ; python_version >= "3.10" and python_version < "4.0"
+numpy==1.26.4 ; python_version >= "3.10" and python_version < "4.0"
+omegaconf==2.3.0 ; python_version >= "3.10" and python_version < "4.0"
+openai==1.21.2 ; python_version >= "3.10" and python_version < "4.0"
+openpyxl==3.1.2 ; python_version >= "3.10" and python_version < "4.0"
+orjson==3.10.1 ; python_version >= "3.10" and python_version < "4.0"
+packaging==23.2 ; python_version >= "3.10" and python_version < "4.0"
+pandas==2.2.2 ; python_version >= "3.10" and python_version < "4.0"
+parso==0.8.4 ; python_version >= "3.10" and python_version < "4.0"
+pathspec==0.12.1 ; python_version >= "3.10" and python_version < "4.0"
+pexpect==4.9.0 ; python_version >= "3.10" and python_version < "4.0" and (sys_platform != "win32" and sys_platform != "emscripten")
+pillow==10.3.0 ; python_version >= "3.10" and python_version < "4.0"
+platformdirs==3.11.0 ; python_version >= "3.10" and python_version < "4.0"
+prompt-toolkit==3.0.43 ; python_version >= "3.10" and python_version < "4.0"
+psutil==5.9.8 ; python_version >= "3.10" and python_version < "4.0"
+ptyprocess==0.7.0 ; python_version >= "3.10" and python_version < "4.0" and (sys_platform != "win32" and sys_platform != "emscripten")
+pure-eval==0.2.2 ; python_version >= "3.10" and python_version < "4.0"
+pycparser==2.22 ; python_version >= "3.10" and python_version < "4.0"
+pydantic-core==2.18.1 ; python_version >= "3.10" and python_version < "4.0"
+pydantic==2.7.0 ; python_version >= "3.10" and python_version < "4.0"
+pydot==2.0.0 ; python_version >= "3.10" and python_version < "4.0"
+pydub==0.25.1 ; python_version >= "3.10" and python_version < "4.0"
+pygit2==1.14.1 ; python_version >= "3.10" and python_version < "4.0"
+pygments==2.17.2 ; python_version >= "3.10" and python_version < "4.0"
+pygtrie==2.5.0 ; python_version >= "3.10" and python_version < "4.0"
+pyparsing==3.1.2 ; python_version >= "3.10" and python_version < "4.0"
+python-dateutil==2.9.0.post0 ; python_version >= "3.10" and python_version < "4.0"
+python-dotenv==1.0.1 ; python_version >= "3.10" and python_version < "4.0"
+python-multipart==0.0.9 ; python_version >= "3.10" and python_version < "4.0"
+pytz==2024.1 ; python_version >= "3.10" and python_version < "4.0"
+pywin32==306 ; python_version >= "3.10" and python_version < "4.0" and sys_platform == "win32"
+pyyaml==6.0.1 ; python_version >= "3.10" and python_version < "4.0"
+pyzmq==26.0.0 ; python_version >= "3.10" and python_version < "4.0"
+referencing==0.34.0 ; python_version >= "3.10" and python_version < "4.0"
+requests==2.31.0 ; python_version >= "3.10" and python_version < "4.0"
+rich==13.7.1 ; python_version >= "3.10" and python_version < "4.0"
+rpds-py==0.18.0 ; python_version >= "3.10" and python_version < "4.0"
+ruamel-yaml-clib==0.2.8 ; platform_python_implementation == "CPython" and python_version < "3.13" and python_version >= "3.10"
+ruamel-yaml==0.18.6 ; python_version >= "3.10" and python_version < "4.0"
+ruff==0.4.1 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
+scmrepo==3.3.1 ; python_version >= "3.10" and python_version < "4.0"
+semantic-version==2.10.0 ; python_version >= "3.10" and python_version < "4.0"
+semver==3.0.2 ; python_version >= "3.10" and python_version < "4.0"
+setuptools==69.5.1 ; python_version >= "3.10" and python_version < "4.0"
+shellingham==1.5.4 ; python_version >= "3.10" and python_version < "4.0"
+shortuuid==1.0.13 ; python_version >= "3.10" and python_version < "4.0"
+shtab==1.7.1 ; python_version >= "3.10" and python_version < "4.0"
+six==1.16.0 ; python_version >= "3.10" and python_version < "4.0"
+smmap==5.0.1 ; python_version >= "3.10" and python_version < "4.0"
+sniffio==1.3.1 ; python_version >= "3.10" and python_version < "4.0"
+sqlalchemy==2.0.29 ; python_version >= "3.10" and python_version < "4.0"
+sqltrie==0.11.0 ; python_version >= "3.10" and python_version < "4.0"
+stack-data==0.6.3 ; python_version >= "3.10" and python_version < "4.0"
+starlette==0.37.2 ; python_version >= "3.10" and python_version < "4.0"
+tabulate==0.9.0 ; python_version >= "3.10" and python_version < "4.0"
+tenacity==8.2.3 ; python_version >= "3.10" and python_version < "4.0"
+tomlkit==0.12.0 ; python_version >= "3.10" and python_version < "4.0"
+toolz==0.12.1 ; python_version >= "3.10" and python_version < "4.0"
+tornado==6.4 ; python_version >= "3.10" and python_version < "4.0"
+tqdm==4.66.2 ; python_version >= "3.10" and python_version < "4.0"
+traitlets==5.14.2 ; python_version >= "3.10" and python_version < "4.0"
+typer==0.12.3 ; python_version >= "3.10" and python_version < "4.0"
+typing-extensions==4.11.0 ; python_version >= "3.10" and python_version < "4.0"
+typing-inspect==0.9.0 ; python_version >= "3.10" and python_version < "4.0"
+tzdata==2024.1 ; python_version >= "3.10" and python_version < "4.0"
+urllib3==2.2.1 ; python_version >= "3.10" and python_version < "4.0"
+uvicorn==0.29.0 ; python_version >= "3.10" and python_version < "4.0" and sys_platform != "emscripten"
+vine==5.1.0 ; python_version >= "3.10" and python_version < "4.0"
+voluptuous==0.14.2 ; python_version >= "3.10" and python_version < "4.0"
+wcwidth==0.2.13 ; python_version >= "3.10" and python_version < "4.0"
+websockets==11.0.3 ; python_version >= "3.10" and python_version < "4.0"
+yarl==1.9.4 ; python_version >= "3.10" and python_version < "4.0"
+zc-lockfile==3.0.post1 ; python_version >= "3.10" and python_version < "4.0"