Update app.py

app.py

@@ -1,217 +1,85 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 import random
-     
-from prompts import GAME_MASTER, COMPRESS_HISTORY, ADJUST_STATS
-def format_prompt(message, history):
-    prompt=""
 
-    prompt = "<s>"
-    
-    for user_prompt, bot_response in history:
-        prompt += f"[INST] {user_prompt} [/INST]"
-        prompt += f" {bot_response}</s> "
+from prompts import GAME_MASTER, COMPRESS_HISTORY, THREE_D_BASE
+
+temperature = 0.92
+top_p = 0.95
+repetition_penalty = 1.05
+MAX_HISTORY = 80
 
+def format_prompt(message, history):
+    prompt = "<s>"
+    for user, bot in history or []:
+        prompt += f"[INST] {user} [/INST] {bot} </s> "
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-
-temperature=0.99
-top_p=0.95
-repetition_penalty=1.0
-
-def compress_history(history,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
+def compress_history(history_str):
     client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
-    
-    print("COMPRESSING")
-    formatted_prompt=f"{COMPRESS_HISTORY.format(history=history)}"
-    generate_kwargs = dict(
-        temperature=temperature,
-        max_new_tokens=1024,
-        top_p=top_p,
-        repetition_penalty=repetition_penalty,
-        do_sample=True,
-        seed=random.randint(1,99999999999)
-        #seed=42,
-    )
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+    prompt = COMPRESS_HISTORY.format(history=history_str)
     output = ""
-    for response in stream:
-        output += response.token.text
+    for chunk in client.text_generation(prompt, temperature=0.7, max_new_tokens=1024, top_p=0.9, stream=True):
+        output += getattr(chunk, 'token', chunk).text
     return output
 
-MAX_HISTORY=100
-opts=[]
-def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
-    opts.clear()
+def generate(msg, history, max_tokens=1024, stats=None, world_style="Epic Unreal Engine"):
     client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
     
-    temperature = float(temperature)
-    if temperature < 1e-2:
-        temperature = 1e-2
-    top_p = float(top_p)
-
-    generate_kwargs = dict(
-        temperature=temperature,
-        max_new_tokens=max_new_tokens,
-        top_p=top_p,
-        repetition_penalty=repetition_penalty,
-        do_sample=True,
-        seed=random.randint(1,99999999999)
-        #seed=42,
-    )
-    cnt=0
-    stats=health
-    history1=history
-    '''
-    stats="*******************\n"
-    for eac in health:
-        stats+=f'{eac}\n'
-    stats+="*******************\n"
-    '''
-    for ea in history:
-        print (ea)
-        for l in ea:
-            print (l)
-            cnt+=len(l.split("\n"))
-    print(f'cnt:: {cnt}')
-    if cnt > MAX_HISTORY:
-        history1 = compress_history(str(history), temperature, top_p, repetition_penalty)
-    formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1,stats=stats,dice=random.randint(1,10))}, {prompt}", history)
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+    stats = stats or "Health: 100\nCreativity: 95\nVision: 88"
+    dice = random.randint(1, 10)
+    
+    gm_text = GAME_MASTER.format(history=history or "New world creation begins...", stats=stats, dice=dice, prompt=msg)
+    full_prompt = format_prompt(gm_text, history)
+    
+    kwargs = {
+        "temperature": temperature,
+        "max_new_tokens": max_tokens,
+        "top_p": top_p,
+        "repetition_penalty": repetition_penalty,
+        "do_sample": True,
+        "seed": random.randint(1, 999999999)
+    }
+    
     output = ""
-
-    for response in stream:
-        output += response.token.text
-        if history:
-            yield [(prompt,output)],stats,None,None
-        else:
-            yield [(prompt,output)],stats,None,None
-    generate_kwargs2 = dict(
-        temperature=temperature,
-        max_new_tokens=128,
-        top_p=top_p,
-        repetition_penalty=repetition_penalty,
-        do_sample=True,
-        seed=random.randint(1,99999999999)
-        #seed=42,
-    )
-    #history=""
-    #formatted_prompt2 = format_prompt(f"{ADJUST_STATS.format(history=output,health=health)}, {prompt}", history)
-    #stream2 = client.text_generation(f"{ADJUST_STATS.format(history=output,health=health)}", **generate_kwargs2, stream=True, details=True, return_full_text=False)  
-    #output2=""
-    #for response in stream2:
-    #    output2 += response.token.text
-       
-    lines = output.strip().strip("\n").split("\n")
-    skills=[]
-    skill_dict={}
-    option_drop=[]
-    new_stat="*******************\n"
-    for i,line in enumerate(lines):
-        if "Choices:" in line:
-            for z in range(1,5):
-                try:
-                    if f'{z}' in lines[i+z]:
-                        print(lines[i+z].split(" ",1)[1])
-                        opts.append(lines[i+z].split(" ",1)[1])
-                except Exception:
-                    pass
-        if ": " in line[:12]:
-            try:
-                lab_1 = line.split(": ")[0]
-                
-                skill_1 = line.split(": ")[1].split(" ")[0].split("<")[0]
-                skill_1=int(skill_1)
-                skill_dict[lab_1]=skill_1
-                #skill ={lab_1:skill_1}
-                
-                new_stat += f'{lab_1}: {skill_1}\n'
-                
-                print(skills)
-            except Exception as e:
-                print (f'--Error :: {e}')
-            print(f'Line:: {line}')
-    skills.append(skill_dict)
-    new_stat+="*******************\n"
-    stats=new_stat
-    option_drop=gr.Dropdown(label="Choices", choices=[e for e in opts])
-
-    if history:
-        history.append((prompt,output))
-        yield history,stats,skills,option_drop
-    else:
-        yield [(prompt,output)],stats,skills,option_drop
+    for chunk in client.text_generation(full_prompt, **kwargs, stream=True, details=True, return_full_text=False):
+        output += chunk.token.text
+        yield [(msg, output)], stats, None
+    
+    # Append 3D prompts
+    three_d = f"\n\n=== 3D GENERATION PROMPTS ===\n{THREE_D_BASE.format(subject='main scene', style=world_style, lighting='volumetric god rays', mood='epic cinematic', camera='wide establishing shot')}\n"
+    three_d += f"1. Main World: {output[:250].replace(chr(10), ' ')} ...\n"
+    three_d += f"2. Key Asset: Character or object from the scene...\n"
+    
+    final = output + three_d
+    
+    if history is None:
+        history = []
+    history.append((msg, final))
+    
+    yield history, stats, gr.Dropdown(choices=["Continue", "New Location", "Generate Asset"], value="Continue")
 
 def clear_fn():
-    return None,None
+    return None, None, None
 
-base_stats=[
-    {},
-]
-text_stats='''*******************
-Health: 100
-Power: 20
-Strength: 24
-Intelligence: 15
-Charisma: 15
-Dexterity: 18
-Wisdom: 15
-Compassion: 15
-Courage: 15
-*******************    
-'''
+with gr.Blocks(title="3D World Architect RPG") as demo:
+    gr.HTML("<center><h1>🌌 3D World Architect RPG</h1><h3>Mixtral 8x7B • Narrative + Production 3D Prompts</h3></center>")
     
-with gr.Blocks() as app:
-    gr.HTML("""<center><h1>Mixtral 8x7B RPG</h1><h3>Role Playing Game Master</h3>""")
-    with gr.Group():
-        with gr.Row():
-            with gr.Column(scale=3):
-                chatbot = gr.Chatbot(label="Mixtral 8x7B Game Master",height=500, layout='panel', show_copy_button=True)
-                with gr.Row():
-                    with gr.Column(scale=3):
-                        opt=gr.Dropdown(label="Choices",choices=["Start a new game"],allow_custom_value=True, value="Start a new game", interactive=True)
-                        #prompt=gr.Textbox(label = "Prompt", value="Start a new game")
-                    with gr.Column(scale=2):
-                        button=gr.Button()
-                    #models_dd=gr.Dropdown(choices=[m for m in return_list],interactive=True)
-                with gr.Row():
-                    stop_button=gr.Button("Stop")
-                    clear_btn = gr.Button("Clear")
-                with gr.Row():
-                    tokens = gr.Slider(label="Max new tokens",value=2096,minimum=0,maximum=1048*10,step=64,interactive=False, visible=False,info="The maximum numbers of new tokens")
-            with gr.Column(scale=1):
-                json_out=gr.JSON(value=base_stats)
-                char_stats=gr.Textbox(value=text_stats)
-                textboxes = []
-                if opts:
-                    textboxes.clear()
-                    for i in range(len(opts)-1):
-                        t = gr.Button(f"{opts[i]}")
-                    textboxes.append(t)
-    #text=gr.JSON()
-    #inp_query.change(search_models,inp_query,models_dd)
-    #test_b=test_btn.click(itt,url,e_box)
-    clear_btn.click(clear_fn,None,[opt,chatbot])
-    go=button.click(generate,[opt,chatbot,tokens,char_stats],[chatbot,char_stats,json_out,opt])
-    stop_button.click(None,None,None,cancels=[go])
-app.launch(show_api=False) 
-
-
-
-'''
-examples=[["Start the Game", None, None, None, None, None, ],
-          ["Start a Game based in the year 1322", None, None, None, None, None,],
-         ]
-
-gr.ChatInterface(
-    fn=generate,
-    chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
-    additional_inputs=additional_inputs,
-    title="Mixtral RPG Game Master",
-    examples=examples,
-    concurrency_limit=20,
-).launch(share=True,show_api=True)
-'''
-
-
+    with gr.Row():
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(height=650, label="World Narrative + 3D Prompts", show_copy_button=True)
+            msg = gr.Textbox(placeholder="Enter your action or describe a new world...", label="Your Command", lines=3)
+            with gr.Row():
+                send = gr.Button("Send", variant="primary")
+                clear = gr.Button("New World")
+        
+        with gr.Column(scale=1):
+            stats = gr.Textbox(value="Health: 100\nCreativity: 95\nVision: 88", label="Architect Stats", lines=12)
+            style = gr.Dropdown(["Epic Unreal Engine", "Photorealistic", "Stylized Fantasy", "Cyberpunk Neon", "Low Poly"], value="Epic Unreal Engine", label="3D Style")
+            tokens = gr.Slider(512, 2048, 1024, label="Max Tokens")
+
+    send.click(generate, [msg, chatbot, tokens, stats, style], [chatbot, stats, msg])
+    clear.click(clear_fn, outputs=[chatbot, msg, stats])
+
+demo.launch(show_api=False)